jit_macros.h 230 B

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