Searched refs:CUDACodeCache (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_cudacodecache.py | 9 from torch._inductor.codecache import CUDACodeCache
43 object_file_path, object_hash_key, source_code_path0 = CUDACodeCache.compile(
46 dll_wrapper, so_hash_key, source_code_path1 = CUDACodeCache.load(
68 CUDACodeCache.compile(error_source_code, "o")
|
| H A D | test_codecache.py | 16 CUDACodeCache,
771 CUDACodeCache.compile("test123.cu", "so", ["-Wsomething"])
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | autotune_process.py | 34 CUDACodeCache,
700 self.hash_key, self.source_file = CUDACodeCache.write(self.source_code, "so")
706 CUDACodeCache.compile(self.source_code, "so")
784 self.DLL, self.hash_key, self.source_file = CUDACodeCache.load(
|
| H A D | async_compile.py | 22 CUDACodeCache,
234 return CUDACodeCache.load(source_code, dst_file_ext)[0]
|
| H A D | codecache.py | 3328 class CUDACodeCache: class
3390 cls.cache[key] = CUDACodeCache.CacheEntry(input_path, output_path)
|
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/cuda/ |
| H A D | cutlass_utils.py | 339 _compile_method_orig = torch._inductor.codecache.CUDACodeCache.compile
|