Searched defs:fn_cache (Results 1 – 2 of 2) sorted by relevance
78 at::cuda::jit::NvrtcFunction &fn_cache, in launch_jitted_unrolled_kernel()113 std::mutex &jiterator_mutex, JittedVecKernelCache &fn_cache, in launch_jitted_vectorized_kernel()
904 std::array<at::cuda::jit::NvrtcFunction, 3> &fn_cache, in launch_jitted_reduce_kernel()1348 …static std::vector<std::array<at::cuda::jit::NvrtcFunction, 3>> fn_cache(c10::cuda::device_count()… in jitted_gpu_reduce_kernel() local