mirror of https://github.com/hpcaitech/ColossalAI
13 lines
238 B
C
13 lines
238 B
C
|
/*This code from NVIDIA apex:
|
||
|
|
* https://github.com/NVIDIA/apex
|
||
|
|
* with minor changes. */
|
||
|
|
|
||
|
|
#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
|