Home
last modified time | relevance | path

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

/external/pytorch/torch/_inductor/codegen/
Dcpp_wrapper_cuda.py381 device_index, call_args = self.prepare_triton_kernel_call(
Dwrapper.py1560 def prepare_triton_kernel_call(self, device_index, call_args): member in WrapperCodeGen
1665 device_index, call_args_str = self.prepare_triton_kernel_call(