apex / csrc /compat.h
camenduru's picture
thanks to NVIDIA ❤
e828767
#ifndef TORCH_CHECK
#define TORCH_CHECK AT_CHECK
#endif
#ifdef VERSION_GE_1_3
#define DATA_PTR data_ptr
#else
#define DATA_PTR data
#endif