where is the script for replacing torch graphs with triton kernels

[JINO-ROHIT]

Honestly just trying to read and learn from this repo, I was wondering if someone can direct/explain me the piece where instead of torch graphs being compiled, the triton kernels kick in.

Thanks very much