mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
29 lines
992 B
C
29 lines
992 B
C
#ifndef THCP_CUDA_MODULE_INC
|
|
#define THCP_CUDA_MODULE_INC
|
|
|
|
extern PyObject *THCPDoubleStorageClass;
|
|
extern PyObject *THCPFloatStorageClass;
|
|
extern PyObject *THCPLongStorageClass;
|
|
extern PyObject *THCPIntStorageClass;
|
|
extern PyObject *THCPHalfStorageClass;
|
|
extern PyObject *THCPShortStorageClass;
|
|
extern PyObject *THCPCharStorageClass;
|
|
extern PyObject *THCPByteStorageClass;
|
|
|
|
extern PyObject *THCPDoubleTensorClass;
|
|
extern PyObject *THCPFloatTensorClass;
|
|
extern PyObject *THCPLongTensorClass;
|
|
extern PyObject *THCPIntTensorClass;
|
|
extern PyObject *THCPHalfTensorClass;
|
|
extern PyObject *THCPShortTensorClass;
|
|
extern PyObject *THCPCharTensorClass;
|
|
extern PyObject *THCPByteTensorClass;
|
|
|
|
extern PyObject * THCPModule_getDevice_wrap(PyObject *self);
|
|
extern PyObject * THCPModule_setDevice_wrap(PyObject *self, PyObject *arg);
|
|
void THCPModule_setDevice(int idx);
|
|
extern PyObject * THCPModule_getDriverVersion(PyObject *self);
|
|
extern PyObject * THCPModule_isDriverSufficient(PyObject *self);
|
|
|
|
#endif
|