Searched refs:parallel_num_threads (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_halide.py | 15 from torch._inductor.utils import parallel_num_threads
83 "parallelism": parallel_num_threads(),
|
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/ |
| H A D | cpp_micro_gemm.py | 13 from ..utils import IndentedBuffer, parallel_num_threads
777 num_threads = parallel_num_threads()
|
| H A D | cpp_gemm_template.py | 16 from ..utils import cache_on_self, has_free_symbols, parallel_num_threads
575 num_threads = parallel_num_threads()
|
| H A D | cpp.py | 40 parallel_num_threads,
1754 "max_threads" if config.cpp.dynamic_threads else parallel_num_threads()
1979 threads = parallel_num_threads()
2597 else parallel_num_threads()
4592 return cls(self.args, parallel_num_threads(), *args)
|
| H A D | halide.py | 41 parallel_num_threads,
1452 "parallelism": parallel_num_threads(),
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | utils.py | 1251 num_threads=parallel_num_threads(),
1500 def parallel_num_threads(): function
1846 and (config.cpp.dynamic_threads or parallel_num_threads() != 1)
|