Caffe2 - C++ API
A deep learning, cross platform ML framework
thnvrtc.cpp
1 #include <torch/csrc/jit/fuser/cuda/thnvrtc.h>
2 #include <iostream>
3 
4 THNVRTC* torch_load_nvrtc() {
5  auto self = new THNVRTC();
6 #define CREATE_ASSIGN(name) self->name = name;
7  TORCH_FORALL_NVRTC(CREATE_ASSIGN)
8  return self;
9 }