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
|