Home
last modified time | relevance | path

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

/external/pytorch/torch/_inductor/codegen/
Dsimd.py1585 def codegen_combo_kernel(self, combo_kernel_node): argument
1586 subkernel_nodes = combo_kernel_node.get_subkernel_nodes()
1587 custom_part_algorithm = combo_kernel_node.use_custom_partition_algo
1588 enable_autotune = combo_kernel_node.enable_autotune
1598 kernel_name = self.define_kernel(src_code, [combo_kernel_node], kernel)
1599 self.codegen_comment([combo_kernel_node])