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 |