Searched refs:cuda_kernel_driver (Results 1 – 3 of 3) sorted by relevance
4 from torch._inductor.codegen.codegen_device_driver import cuda_kernel_driver37 header = cuda_kernel_driver()
18 from .codegen_device_driver import cuda_kernel_driver, cuda_kernel_header187 self.header.splice(maybe_hipify_code_wrapper(cuda_kernel_driver()))
7 def cuda_kernel_driver() -> str: function