Searched refs:CompiledKernel (Results 1 – 3 of 3) sorted by relevance
175 from triton.compiler import CompiledKernel185 CompiledKernel.launch_enter_hook = launch_enter_hook186 CompiledKernel.launch_exit_hook = launch_exit_hook
52 def CompiledKernel(fn, *inputs, **kwargs): function110 compiled_op = CompiledKernel(
56 from triton.compiler import CompiledKernel468 "launch_enter_hook": CompiledKernel.launch_enter_hook,469 "launch_exit_hook": CompiledKernel.launch_exit_hook,