1 #ifndef THCP_CUDA_MODULE_INC 2 #define THCP_CUDA_MODULE_INC 4 extern THCState *state;
7 void THCPModule_setDevice(
int idx);
8 PyObject * THCPModule_getDevice_wrap(PyObject *
self);
9 PyObject * THCPModule_setDevice_wrap(PyObject *
self, PyObject *arg);
10 PyObject * THCPModule_getDeviceName_wrap(PyObject *
self, PyObject *arg);
11 PyObject * THCPModule_getDriverVersion(PyObject *
self);
12 PyObject * THCPModule_isDriverSufficient(PyObject *
self);
13 PyObject * THCPModule_getCurrentBlasHandle_wrap(PyObject *
self);
1.8.11
