Home
last modified time | relevance | path

Searched defs:can_fuse_vertical (Results 1 – 8 of 8) sorted by relevance

/external/pytorch/torch/_inductor/codegen/rocm/
Drocm_cpp_scheduling.py39 def can_fuse_vertical( member in ROCmCPPScheduling
/external/pytorch/torch/_inductor/codegen/cuda/
Dcuda_cpp_scheduling.py44 def can_fuse_vertical( member in CUDACPPScheduling
/external/pytorch/test/inductor/extension_backends/cpp/
Dextension_codegen_backend.py21 def can_fuse_vertical(self, node1, node2): member in ExtensionScheduling
/external/pytorch/test/inductor/extension_backends/triton/
Dextension_codegen_backend.py16 def can_fuse_vertical(self, node1, node2): member in ExtensionScheduling
/external/pytorch/torch/_inductor/codegen/
Dcuda_combined_scheduling.py43 def can_fuse_vertical(self, node1: BaseSchedulerNode, node2: BaseSchedulerNode): member in CUDACombinedScheduling
Dsimd.py1062 can_fuse_vertical = can_fuse variable in SIMDScheduling
Dcpp.py4148 def can_fuse_vertical(self, node1, node2): member in CppScheduling
/external/pytorch/torch/_inductor/
Dscheduler.py2996 def can_fuse_vertical( member in Scheduler
3598 def can_fuse_vertical( member in BaseScheduling