提交 9618fc79 authored 作者: abergeron's avatar abergeron

Merge pull request #2388 from nouiz/win64_gpu_fix

Fix truncated gpu dev pointer on windows 64 bits.
...@@ -2731,13 +2731,13 @@ CudaNdarray_get_dev_data(CudaNdarray *self, void *closure) ...@@ -2731,13 +2731,13 @@ CudaNdarray_get_dev_data(CudaNdarray *self, void *closure)
{ {
float * p = CudaNdarray_DEV_DATA(self); float * p = CudaNdarray_DEV_DATA(self);
//printf("get_dev_data %p %li \n", p, (long int)p ); //printf("get_dev_data %p %li \n", p, (long int)p );
return PyInt_FromLong((long int) CudaNdarray_DEV_DATA(self)); return PyInt_FromSize_t((size_t) CudaNdarray_DEV_DATA(self));
} }
static int static int
CudaNdarray_set_dev_data(CudaNdarray *self, PyObject *value, void *closure) CudaNdarray_set_dev_data(CudaNdarray *self, PyObject *value, void *closure)
{ {
long int newdevdata = PyInt_AsLong(value); Py_ssize_t newdevdata = PyInt_AsSsize_t(value);
//printf("set_dev_data %p %li \n",(float*)newdevdata ,newdevdata); //printf("set_dev_data %p %li \n",(float*)newdevdata ,newdevdata);
if (PyErr_Occurred()) if (PyErr_Occurred())
{ {
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论