Home
last modified time | relevance | path

Searched refs:CUDACodeCache (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_cudacodecache.py9 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 Dtest_codecache.py16 CUDACodeCache,
771 CUDACodeCache.compile("test123.cu", "so", ["-Wsomething"])
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dautotune_process.py34 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 Dasync_compile.py22 CUDACodeCache,
234 return CUDACodeCache.load(source_code, dst_file_ext)[0]
H A Dcodecache.py3328 class CUDACodeCache: class
3390 cls.cache[key] = CUDACodeCache.CacheEntry(input_path, output_path)
/aosp_15_r20/external/pytorch/torch/_inductor/codegen/cuda/
H A Dcutlass_utils.py339 _compile_method_orig = torch._inductor.codecache.CUDACodeCache.compile