mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-08-11 04:41:54 +00:00
079bf3cb26
* [misc] update pre-commit * [misc] run pre-commit * [misc] remove useless configuration files * [misc] ignore cuda for clang-format
11 lines
214 B
C
11 lines
214 B
C
// modified from https://github.com/NVIDIA/apex/blob/master/csrc/compat.h
|
|
#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
|