Searched refs:codegen_node (Results 1 – 7 of 7) sorted by relevance
| /external/pytorch/test/inductor/extension_backends/triton/ |
| D | extension_codegen_backend.py | 28 def codegen_node(self, node): member in ExtensionScheduling
29 self._triton_scheduling.codegen_node(node)
|
| /external/pytorch/test/inductor/extension_backends/cpp/ |
| D | extension_codegen_backend.py | 33 def codegen_node(self, node): member in ExtensionScheduling
34 self._scheduling.codegen_node(node)
|
| /external/pytorch/torchgen/_autoheuristic/ |
| D | ah_tree.py | 201 def codegen_node(node: DecisionTreeNode, depth: int) -> None: function
220 codegen_node(node.left, depth + 1)
223 codegen_node(node.right, depth + 1)
262 codegen_node(self.root, 1)
|
| /external/pytorch/torch/_inductor/codegen/ |
| D | cuda_combined_scheduling.py | 79 def codegen_node(self, node: Union[FusedSchedulerNode, SchedulerNode]): member in CUDACombinedScheduling
80 return self._triton_scheduling.codegen_node(node)
|
| D | simd.py | 1136 def codegen_node( member in SIMDScheduling
|
| D | cpp.py | 4454 def codegen_node( member in CppScheduling
|
| /external/pytorch/torch/_inductor/ |
| D | scheduler.py | 3461 self.get_backend(device).codegen_node(node)
3646 def codegen_node(self, node: Union[FusedSchedulerNode, SchedulerNode]) -> None: member in BaseScheduling
|