Caffe2 - C++ API
A deep learning, cross platform ML framework
Storage.h
1 #ifndef THCP_STORAGE_INC
2 #define THCP_STORAGE_INC
3 
4 #define THCPStorageStr TH_CONCAT_STRING_3(torch.cuda.,Real,Storage)
5 #define THCPStorageClass TH_CONCAT_3(THCP,Real,StorageClass)
6 #define THCPStorage_(NAME) TH_CONCAT_4(THCP,Real,Storage_,NAME)
7 
8 #define THCPDoubleStorage_Check(obj) \
9  PyObject_IsInstance(obj, THCPDoubleStorageClass)
10 #define THCPFloatStorage_Check(obj) \
11  PyObject_IsInstance(obj, THCPFloatStorageClass)
12 #define THCPHalfStorage_Check(obj) \
13  PyObject_IsInstance(obj, THCPHalfStorageClass)
14 #define THCPLongStorage_Check(obj) \
15  PyObject_IsInstance(obj, THCPLongStorageClass)
16 #define THCPIntStorage_Check(obj) \
17  PyObject_IsInstance(obj, THCPIntStorageClass)
18 #define THCPShortStorage_Check(obj) \
19  PyObject_IsInstance(obj, THCPShortStorageClass)
20 #define THCPCharStorage_Check(obj) \
21  PyObject_IsInstance(obj, THCPCharStorageClass)
22 #define THCPByteStorage_Check(obj) \
23  PyObject_IsInstance(obj, THCPByteStorageClass)
24 #define THCPBoolStorage_Check(obj) \
25  PyObject_IsInstance(obj, THCPBoolStorageClass)
26 
27 #define THCPDoubleStorage_CData(obj) (obj)->cdata
28 #define THCPFloatStorage_CData(obj) (obj)->cdata
29 #define THCPLongStorage_CData(obj) (obj)->cdata
30 #define THCPIntStorage_CData(obj) (obj)->cdata
31 #define THCPShortStorage_CData(obj) (obj)->cdata
32 #define THCPCharStorage_CData(obj) (obj)->cdata
33 #define THCPByteStorage_CData(obj) (obj)->cdata
34 #define THCPBoolStorage_CData(obj) (obj)->cdata
35 
36 #ifdef _THP_CORE
37 #define THCPStorageType TH_CONCAT_3(THCP,Real,StorageType)
38 #define THCPStorageBaseStr TH_CONCAT_STRING_3(Cuda,Real,StorageBase)
39 #endif
40 
41 #include <torch/csrc/cuda/override_macros.h>
42 
43 #define THC_GENERIC_FILE "torch/csrc/generic/Storage.h"
44 #include <THC/THCGenerateAllTypes.h>
45 
46 #define THC_GENERIC_FILE "torch/csrc/generic/Storage.h"
47 #include <THC/THCGenerateBoolType.h>
48 
49 #endif