Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/_inductor/codegen/
H A Dcpp.py1989 loop_nest.max_parallel_depth(), threads
1993 loop_nest.max_parallel_depth(), threads
2156 def decide_parallel_depth(self, max_parallel_depth, threads): argument
2158 ranges = self.call_ranges[:max_parallel_depth]
3863 def decide_parallel_depth(self, max_parallel_depth, threads) -> int: argument
3878 max_parallel_depth,
4917 def max_parallel_depth(self): member in LoopNestWithSplit
4946 par_depth <= self.max_parallel_depth()