FM / cuda

/usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/cuda/

/usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/cuda

Name	Size	Mode	Actions
comm.h	1525	0644	edit dl rm
device_set.h	152	0644	edit dl rm
Event.h	487	0644	edit dl rm
Module.h	503	0644	edit dl rm
nccl.h	5037	0644	edit dl rm
override_macros.h	1887	0644	edit dl rm
python_comm.h	114	0644	edit dl rm
python_nccl.h	616	0644	edit dl rm
restore_macros.h	820	0644	edit dl rm
serialization.h	524	0644	edit dl rm
Storage.h	2585	0644	edit dl rm
Stream.h	524	0644	edit dl rm
THCP.h	421	0644	edit dl rm
undef_macros.h	932	0644	edit dl rm
utils.h	871	0644	edit dl rm

Edit: /usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/cuda/Storage.h (2585B)

#ifndef THCP_STORAGE_INC
#define THCP_STORAGE_INC

#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 THCPBoolStorage_Check(obj) \
    PyObject_IsInstance(obj, THCPBoolStorageClass)
#define THCPBFloat16Storage_Check(obj) \
    PyObject_IsInstance(obj, THCPBFloat16StorageClass)
#define THCPComplexDoubleStorage_Check(obj) \
    PyObject_IsInstance(obj, THCPDoubleStorageClass)
#define THCPComplexFloatStorage_Check(obj) \
    PyObject_IsInstance(obj, THCPFloatStorageClass)

#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
#define THCPBoolStorage_CData(obj)               (obj)->cdata
#define THCPBFloat16Storage_CData(obj)           (obj)->cdata
#define THCPComplexDoubleStorage_CData(obj)      (obj)->cdata
#define THCPComplexFloatStorage_CData(obj)       (obj)->cdata

#define THCPStorageType TH_CONCAT_3(THCP,Real,StorageType)
#define THCPStorageBaseStr TH_CONCAT_STRING_3(Cuda,Real,StorageBase)

#include <torch/csrc/cuda/override_macros.h>

#define THC_GENERIC_FILE "torch/csrc/generic/Storage.h"
#include <THC/THCGenerateAllTypes.h>

#define THC_GENERIC_FILE "torch/csrc/generic/Storage.h"
#include <THC/THCGenerateComplexTypes.h>

#define THC_GENERIC_FILE "torch/csrc/generic/Storage.h"
#include <THC/THCGenerateBoolType.h>

#define THC_GENERIC_FILE "torch/csrc/generic/Storage.h"
#include <THC/THCGenerateBFloat16Type.h>

#endif

#ifndef THCP_STORAGE_INC #define THCP_STORAGE_INC #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 THCPBoolStorage_Check(obj) \ PyObject_IsInstance(obj, THCPBoolStorageClass) #define THCPBFloat16Storage_Check(obj) \ PyObject_IsInstance(obj, THCPBFloat16StorageClass) #define THCPComplexDoubleStorage_Check(obj) \ PyObject_IsInstance(obj, THCPDoubleStorageClass) #define THCPComplexFloatStorage_Check(obj) \ PyObject_IsInstance(obj, THCPFloatStorageClass) #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 #define THCPBoolStorage_CData(obj) (obj)->cdata #define THCPBFloat16Storage_CData(obj) (obj)->cdata #define THCPComplexDoubleStorage_CData(obj) (obj)->cdata #define THCPComplexFloatStorage_CData(obj) (obj)->cdata #define THCPStorageType TH_CONCAT_3(THCP,Real,StorageType) #define THCPStorageBaseStr TH_CONCAT_STRING_3(Cuda,Real,StorageBase) #include #define THC_GENERIC_FILE "torch/csrc/generic/Storage.h" #include #define THC_GENERIC_FILE "torch/csrc/generic/Storage.h" #include #define THC_GENERIC_FILE "torch/csrc/generic/Storage.h" #include #define THC_GENERIC_FILE "torch/csrc/generic/Storage.h" #include #endif