Home
last modified time | relevance | path

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

/external/pytorch/torch/_inductor/codegen/
Dtriton_combo_kernel.py87 and not (kernel_map[n].persistent_reduction and kernel_map[n].no_x_dim)
522 if tree.prefix == "r" and sub_kernel.persistent_reduction:
573 if sub_kernel.persistent_reduction:
750 not sub_kernel.inside_reduction or sub_kernel.persistent_reduction
Dtriton.py1287 and self.persistent_reduction
1969 if self.persistent_reduction:
2234 if not self.persistent_reduction:
2270 if not self.persistent_reduction:
2313 self.persistent_reduction
2522 if self.persistent_reduction:
2695 if tree.prefix == "r" and self.persistent_reduction:
2787 if tree.prefix == "r" and self.persistent_reduction:
2909 if tree.prefix == "r" and self.persistent_reduction:
Dsimd.py336 self.persistent_reduction: bool = (
388 is_loop=is_reduction and not self.persistent_reduction,
485 if not self.inside_reduction or self.persistent_reduction:
1344 if kernel.persistent_reduction and config.triton.multi_kernel and not has_sort:
/external/pytorch/torch/_inductor/runtime/
Dtriton_heuristics.py1535 def persistent_reduction( function