Searched refs:CudaKernelParamCache (Results 1 – 3 of 3) sorted by relevance
14 from ..codecache import CudaKernelParamCache47 params = CudaKernelParamCache.get(self.kernel_name)97 params = CudaKernelParamCache.get(self.kernel_name)129 params = CudaKernelParamCache.get(self.kernel_name)
766 from torch._inductor.codecache import CudaKernelParamCache770 CudaKernelParamCache.set(key, params, binary, bin_type)
1954 class CudaKernelParamCache: class