Searched refs:MultiKernelCall (Results 1 – 3 of 3) sorted by relevance
11 from torch._inductor.codegen.multi_kernel import MultiKernelCall115 orig_run = MultiKernelCall.run125 MultiKernelCall, "run", mock_run127 MultiKernelCall,
207 from torch._inductor.codegen.multi_kernel import MultiKernelCall210 return MultiKernelCall(*args, **kwargs)
176 picked_kernel = MultiKernelCall.lookup_choice(kernel_name)229 class MultiKernelCall: class