Home
last modified time | relevance | path

Searched refs:nvrtcCreateProgram (Results 1 – 5 of 5) sorted by relevance

/external/pytorch/aten/src/ATen/cuda/nvrtc_stub/
DATenNVRTC.h37 _(nvrtcCreateProgram) \
108 _(nvrtcCreateProgram) \
/external/pytorch/aten/src/ATen/cuda/detail/
DLazyNVRTC.cpp133 nvrtcResult nvrtcCreateProgram(nvrtcProgram *prog, in nvrtcCreateProgram() function
139 auto fn = reinterpret_cast<decltype(&nvrtcCreateProgram)>(getNVRTCLibrary().sym(__func__)); in nvrtcCreateProgram()
142 lazyNVRTC.nvrtcCreateProgram = fn; in nvrtcCreateProgram()
/external/pytorch/torch/csrc/jit/codegen/fuser/cuda/
Dfused_kernel.cpp120 AT_CUDA_NVRTC_CHECK(nvrtc().nvrtcCreateProgram( in FusedKernelCUDA()
/external/pytorch/aten/src/ATen/native/cuda/
Djit_utils.cpp1535 AT_CUDA_NVRTC_CHECK(nvrtc.nvrtcCreateProgram( in jit_pwise_function()
/external/pytorch/torch/csrc/jit/tensorexpr/
Dcuda_codegen.cpp1281 AT_CUDA_NVRTC_CHECK(nvrtc().nvrtcCreateProgram( in CompileToNVRTC()