Home
last modified time | relevance | path

Searched refs:_module_to_triton_kernel (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_inductor/runtime/
H A Dcompile_tasks.py13 return _module_to_triton_kernel(reload_module(), kernel_name)
16 def _module_to_triton_kernel(mod, kernel_name): function
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dcodecache.py55 _module_to_triton_kernel,
3123 return _module_to_triton_kernel(PyCodeCache.load(source_code), kernel_name)