Searched refs:should_use_persistent_reduction (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/ | ||
| H A D | triton_split_scan.py | 48 def should_use_persistent_reduction(self) -> bool: member in TritonSplitScanKernel |
| H A D | simd.py | 339 else self.should_use_persistent_reduction() |
| H A D | triton.py | 1264 def should_use_persistent_reduction(self) -> bool: member in TritonKernel |