Searched refs:PyCodeCache (Results 1 – 14 of 14) sorted by relevance
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_triton_wrapper.py | 8 from torch._inductor.codecache import PyCodeCache
16 for v in PyCodeCache.cache.values():
|
| H A D | test_kernel_benchmark.py | 12 from torch._inductor.codecache import PyCodeCache
34 PyCodeCache.cache.clear()
38 for v in PyCodeCache.cache.values():
|
| H A D | test_codecache.py | 19 PyCodeCache,
154 for m in torch._inductor.codecache.PyCodeCache.cache.values():
844 self.assertEqual(len(PyCodeCache.cache.keys()), 0)
850 self.assertEqual(len(PyCodeCache.cache.keys()), 0)
|
| H A D | test_multi_kernel.py | 64 codecache.PyCodeCache.cache_clear()
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | wrapper_benchmark.py | 77 from torch._inductor.codecache import PyCodeCache
80 for kernel_key, kernel_mod in PyCodeCache.cache.items():
|
| H A D | metrics.py | 274 from .codecache import PyCodeCache
277 mod = PyCodeCache.load(kernel_module_code)
|
| H A D | autotune_process.py | 37 PyCodeCache,
631 mod = PyCodeCache.load_by_key_path(self.module_cache_key, self.module_path)
673 mod = PyCodeCache.load_by_key_path(self.module_cache_key, self.module_path)
|
| H A D | graph.py | 1681 from .codecache import PyCodeCache
1690 key, path = PyCodeCache.write(code)
1705 mod = PyCodeCache.load_by_key_path(
|
| H A D | select_algorithm.py | 31 from .codecache import code_hash, PersistentCache, PyCodeCache
697 mod = PyCodeCache.load(code, extra)
|
| H A D | codecache.py | 875 graph.current_callable = PyCodeCache.load_by_key_path(
3042 class PyCodeCache: class
3123 return _module_to_triton_kernel(PyCodeCache.load(source_code), kernel_name)
|
| /aosp_15_r20/external/pytorch/torch/profiler/ |
| H A D | profiler.py | 846 from torch._inductor.codecache import PyCodeCache as PyCodeCache unknown
850 for v in PyCodeCache.cache.values()
|
| /aosp_15_r20/external/pytorch/torch/_inductor/runtime/ |
| H A D | compile_tasks.py | 29 return codecache.PyCodeCache.load_by_key_path(key, path)
|
| /aosp_15_r20/external/pytorch/test/dynamo/ |
| H A D | test_aot_autograd_cache.py | 55 for m in torch._inductor.codecache.PyCodeCache.cache.values():
57 torch._inductor.codecache.PyCodeCache.cache_clear()
|
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/ |
| H A D | triton.py | 38 from ..codecache import code_hash, get_path, PyCodeCache
3067 mod = PyCodeCache.load(src_code)
3177 mod = PyCodeCache.load(src_code)
|