Searched refs:template_node (Results 1 – 12 of 12) sorted by relevance
| /external/pytorch/torch/_inductor/codegen/rocm/ |
| D | rocm_cpp_scheduling.py | 76 template_node: BaseSchedulerNode,
83 template_node
85 template_node = cast(SchedulerNode, template_node)
86 _, (numel, rnumel) = template_node.group
88 ctb: ROCmTemplateBuffer = cast(ROCmTemplateBuffer, template_node.node)
91 template_node.mark_run()
95 node_schedule = [template_node]
|
| D | rocm_template.py | 107 template_node: ROCmTemplateBuffer,
116 template_buffer_node=template_node,
|
| /external/pytorch/torch/_inductor/codegen/cuda/ |
| D | cuda_cpp_scheduling.py | 81 template_node: BaseSchedulerNode,
89 template_node
91 template_node = cast(SchedulerNode, template_node)
92 _, (numel, rnumel) = template_node.group
94 ctb: CUDATemplateBuffer = cast(CUDATemplateBuffer, template_node.node)
97 template_node.mark_run()
101 node_schedule = [template_node]
|
| D | cuda_template.py | 108 template_node: CUDATemplateBuffer,
117 template_buffer_node=template_node,
|
| /external/pytorch/torch/_inductor/codegen/ |
| D | cuda_combined_scheduling.py | 61 template_node: BaseSchedulerNode,
64 if self._cuda_cpp_scheduling.is_cuda_cpp_template(template_node):
67 template_node, epilogue_nodes
69 elif self._rocm_cpp_scheduling.is_rocm_cpp_template(template_node):
72 template_node, epilogue_nodes
76 template_node, epilogue_nodes
|
| D | cpp_template.py | 84 template_node: ir.CppTemplateBuffer,
94 template_buffer_node=template_node,
|
| D | simd.py | 1452 self, template_node, epilogue_nodes, only_gen_src_code=False argument
1459 _, (numel, rnumel) = template_node.group
1461 kernel, render = template_node.node.make_kernel_render(template_node.node)
1464 for node in [template_node, *epilogue_nodes]:
1483 node_schedule = [template_node, *epilogue_nodes]
1502 kernel.call_kernel(kernel_name, template_node.node)
1807 template_node = nodes[0]
1812 template_node, epilogue_nodes, only_gen_src_code=True
|
| D | cpp.py | 4483 template_node: BaseSchedulerNode,
4491 template_node
4493 template_node = cast(SchedulerNode, template_node)
4494 _, (_, rnumel) = template_node.group
4496 ctb: ir.CppTemplateBuffer = cast(ir.CppTemplateBuffer, template_node.node)
4516 ctb, template_node.outputs_by_name, epilogue_ir_nodes
4524 for node in [template_node, *epilogue_nodes]:
4529 node_schedule = [template_node, *epilogue_nodes]
|
| /external/python/mako/mako/ext/ |
| D | extract.py | 19 template_node = lexer.Lexer(
22 yield from self.extract_nodes(template_node.get_children())
|
| /external/pytorch/test/inductor/extension_backends/triton/ |
| D | extension_codegen_backend.py | 25 def codegen_template(self, template_node, epilogue_nodes): argument
|
| /external/pytorch/test/inductor/extension_backends/cpp/ |
| D | extension_codegen_backend.py | 30 def codegen_template(self, template_node, epilogue_nodes): argument
|
| /external/pytorch/torch/_inductor/ |
| D | scheduler.py | 3635 template_node: BaseSchedulerNode,
|