xref: /aosp_15_r20/external/pytorch/torch/csrc/THP.h (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1*da0073e9SAndroid Build Coastguard Worker #ifndef THP_H
2*da0073e9SAndroid Build Coastguard Worker #define THP_H
3*da0073e9SAndroid Build Coastguard Worker 
4*da0073e9SAndroid Build Coastguard Worker #include <torch/csrc/Export.h>
5*da0073e9SAndroid Build Coastguard Worker #include <torch/csrc/python_headers.h>
6*da0073e9SAndroid Build Coastguard Worker 
7*da0073e9SAndroid Build Coastguard Worker // Back-compatibility macros, Thanks to http://cx-oracle.sourceforge.net/
8*da0073e9SAndroid Build Coastguard Worker // define PyInt_* macros for Python 3.x.  NB: We must include Python.h first,
9*da0073e9SAndroid Build Coastguard Worker // otherwise we'll incorrectly conclude PyInt_Check isn't defined!
10*da0073e9SAndroid Build Coastguard Worker #ifndef PyInt_Check
11*da0073e9SAndroid Build Coastguard Worker #define PyInt_Check PyLong_Check
12*da0073e9SAndroid Build Coastguard Worker #define PyInt_FromLong PyLong_FromLong
13*da0073e9SAndroid Build Coastguard Worker #define PyInt_AsLong PyLong_AsLong
14*da0073e9SAndroid Build Coastguard Worker #define PyInt_Type PyLong_Type
15*da0073e9SAndroid Build Coastguard Worker #endif
16*da0073e9SAndroid Build Coastguard Worker 
17*da0073e9SAndroid Build Coastguard Worker #include <torch/csrc/Exceptions.h>
18*da0073e9SAndroid Build Coastguard Worker #include <torch/csrc/Generator.h>
19*da0073e9SAndroid Build Coastguard Worker #include <torch/csrc/Module.h>
20*da0073e9SAndroid Build Coastguard Worker #include <torch/csrc/Size.h>
21*da0073e9SAndroid Build Coastguard Worker #include <torch/csrc/Storage.h>
22*da0073e9SAndroid Build Coastguard Worker #include <torch/csrc/Types.h>
23*da0073e9SAndroid Build Coastguard Worker #include <torch/csrc/utils.h> // This requires defined Storage and Tensor types
24*da0073e9SAndroid Build Coastguard Worker #include <torch/csrc/utils/byte_order.h>
25*da0073e9SAndroid Build Coastguard Worker 
26*da0073e9SAndroid Build Coastguard Worker #include <torch/csrc/serialization.h>
27*da0073e9SAndroid Build Coastguard Worker 
28*da0073e9SAndroid Build Coastguard Worker #include <torch/csrc/autograd/python_autograd.h>
29*da0073e9SAndroid Build Coastguard Worker 
30*da0073e9SAndroid Build Coastguard Worker #endif
31