1 #include <ATen/cuda/nvrtc_stub/ATenNVRTC.h> 2 #include <iostream> 3 4 namespace at { namespace cuda { 5 load_nvrtc()6 NVRTC* load_nvrtc() { 7 auto self = new NVRTC(); 8 #define CREATE_ASSIGN(name) self->name = name; 9 AT_FORALL_NVRTC(CREATE_ASSIGN) 10 return self; 11 } 12 13 }} // at::cuda 14