Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_max_autotune.py21 AlgorithmSelectorCache,
92 mat1 = AlgorithmSelectorCache.benchmark_example_value(buf1)
93 mat2 = AlgorithmSelectorCache.benchmark_example_value(buf2)
94 mat3 = AlgorithmSelectorCache.benchmark_example_value(buf3)
95 mat4 = AlgorithmSelectorCache.benchmark_example_value(buf4)
97 out = AlgorithmSelectorCache.benchmark_example_value(layout)
130 mat1 = AlgorithmSelectorCache.benchmark_example_value(buf1)
131 mat2 = AlgorithmSelectorCache.benchmark_example_value(buf2)
132 mat3 = AlgorithmSelectorCache.benchmark_example_value(buf3)
133 mat4 = AlgorithmSelectorCache.benchmark_example_value(buf4)
[all …]
H A Dtest_select_algorithm.py32 patch.object(select_algorithm.AlgorithmSelectorCache, "lookup", skip_cache),
H A Dtest_cpu_select_algorithm.py78 patch.object(select_algorithm.AlgorithmSelectorCache, "lookup", skip_cache),
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dselect_algorithm.py1138 return repr([AlgorithmSelectorCache.key_of(x) for x in input_nodes])
1154 class AlgorithmSelectorCache(PersistentCache): class
1651 return AlgorithmSelectorCache.generate_example_value(
1711 _ALGORITHM_SELECTOR_CACHE: Optional[AlgorithmSelectorCache] = None
1717 _ALGORITHM_SELECTOR_CACHE = AlgorithmSelectorCache()
1732 _ALGORITHM_SELECTOR_CACHE = AlgorithmSelectorCache()