1 #pragma once 2 #include <ATen/cuda/CUDAConfig.h> 3 #include <string> 4 5 // AT_USE_JITERATOR(), controls whether we jit some elementwise kernels 6 #define AT_USE_JITERATOR() true 7 #define jiterator_stringify(...) std::string(#__VA_ARGS__); 8
1 #pragma once 2 #include <ATen/cuda/CUDAConfig.h> 3 #include <string> 4 5 // AT_USE_JITERATOR(), controls whether we jit some elementwise kernels 6 #define AT_USE_JITERATOR() true 7 #define jiterator_stringify(...) std::string(#__VA_ARGS__); 8