| #ifndef THCP_STORAGE_INC |
| #define THCP_STORAGE_INC |
| |
| #define THCPStorage TH_CONCAT_3(THCP,Real,Storage) |
| #define THCPStorageStr TH_CONCAT_STRING_3(torch.cuda.,Real,Storage) |
| #define THCPStorageClass TH_CONCAT_3(THCP,Real,StorageClass) |
| #define THCPStorage_(NAME) TH_CONCAT_4(THCP,Real,Storage_,NAME) |
| |
| #define THCPDoubleStorage_Check(obj) \ |
| PyObject_IsInstance(obj, THCPDoubleStorageClass) |
| #define THCPFloatStorage_Check(obj) \ |
| PyObject_IsInstance(obj, THCPFloatStorageClass) |
| #define THCPHalfStorage_Check(obj) \ |
| PyObject_IsInstance(obj, THCPHalfStorageClass) |
| #define THCPLongStorage_Check(obj) \ |
| PyObject_IsInstance(obj, THCPLongStorageClass) |
| #define THCPIntStorage_Check(obj) \ |
| PyObject_IsInstance(obj, THCPIntStorageClass) |
| #define THCPShortStorage_Check(obj) \ |
| PyObject_IsInstance(obj, THCPShortStorageClass) |
| #define THCPCharStorage_Check(obj) \ |
| PyObject_IsInstance(obj, THCPCharStorageClass) |
| #define THCPByteStorage_Check(obj) \ |
| PyObject_IsInstance(obj, THCPByteStorageClass) |
| |
| #define THCPDoubleStorage_CData(obj) (obj)->cdata |
| #define THCPFloatStorage_CData(obj) (obj)->cdata |
| #define THCPLongStorage_CData(obj) (obj)->cdata |
| #define THCPIntStorage_CData(obj) (obj)->cdata |
| #define THCPShortStorage_CData(obj) (obj)->cdata |
| #define THCPCharStorage_CData(obj) (obj)->cdata |
| #define THCPByteStorage_CData(obj) (obj)->cdata |
| |
| #ifdef _THP_CORE |
| #define THCPStorageType TH_CONCAT_3(THCP,Real,StorageType) |
| #define THCPStorageBaseStr TH_CONCAT_STRING_3(Cuda,Real,StorageBase) |
| #endif |
| |
| #include "override_macros.h" |
| |
| #define THC_GENERIC_FILE "torch/csrc/generic/Storage.h" |
| #include <THC/THCGenerateAllTypes.h> |
| |
| #endif |