1 | #ifndef THCP_CUDA_MODULE_INC |
---|---|
2 | #define THCP_CUDA_MODULE_INC |
3 | |
4 | void THCPModule_setDevice(int idx); |
5 | PyObject* THCPModule_getDevice_wrap(PyObject* self); |
6 | PyObject* THCPModule_setDevice_wrap(PyObject* self, PyObject* arg); |
7 | PyObject* THCPModule_getDeviceName_wrap(PyObject* self, PyObject* arg); |
8 | PyObject* THCPModule_getDriverVersion(PyObject* self); |
9 | PyObject* THCPModule_isDriverSufficient(PyObject* self); |
10 | PyObject* THCPModule_getCurrentBlasHandle_wrap(PyObject* self); |
11 | |
12 | #endif |
13 |