Searched defs:NvrtcFunction (Results 1 – 2 of 2) sorted by relevance
16 struct NvrtcFunction { struct17 CUmodule module = CUmodule();18 CUfunction function = nullptr;
904 std::array<at::cuda::jit::NvrtcFunction, 3> &fn_cache, in launch_jitted_reduce_kernel()