generate_kernel_code_from_nodes (reference) in projects: aosp_15_r20

Searched refs:generate_kernel_code_from_nodes (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_inductor/codegen/
H A D	cuda_combined_scheduling.py	`94 def generate_kernel_code_from_nodes(self, nodes, benchmark_kernel=False): member in CUDACombinedScheduling 95 return self._triton_scheduling.generate_kernel_code_from_nodes(`
H A D	simd.py	`1770 def generate_kernel_code_from_nodes(self, nodes, benchmark_kernel=False): member in SIMDScheduling`
H A D	triton.py	`3064 src_code = self.generate_kernel_code_from_nodes(`
/aosp_15_r20/external/pytorch/torch/_inductor/
H A D	scheduler.py	`3722 triton_code = backend.generate_kernel_code_from_nodes(snodes).strip()`