提交 7e01c09f authored 作者: Micah Bojrab's avatar Micah Bojrab

Updates to solve potential problems with compiler-specific casting

上级 6cc53c22
...@@ -2917,7 +2917,7 @@ GetDeviceMemInfo(PyObject* _unused, PyObject* dummy) ...@@ -2917,7 +2917,7 @@ GetDeviceMemInfo(PyObject* _unused, PyObject* dummy)
cudaGetErrorString(err)); cudaGetErrorString(err));
return NULL; return NULL;
} }
return PyTuple_Pack(2, PyLong_FromLong(free), PyLong_FromLong(total)); return PyTuple_Pack(2, PyLong_FromSize_t(free), PyLong_FromSize_t(total));
} }
/* /*
...@@ -2986,8 +2986,8 @@ CudaNdarray_select_a_gpu(PyObject* _unused, PyObject* dummy) ...@@ -2986,8 +2986,8 @@ CudaNdarray_select_a_gpu(PyObject* _unused, PyObject* dummy)
PyObject * PyObject *
GetTheanoAllocInfo(PyObject* _unused, PyObject* dummy) GetTheanoAllocInfo(PyObject* _unused, PyObject* dummy)
{ {
PyObject* a = PyLong_FromLong(_allocated_size); PyObject* a = PyLong_FromSize_t(_allocated_size);
PyObject* b = PyLong_FromLong(_max_allocated_size); PyObject* b = PyLong_FromSize_t(_max_allocated_size);
PyObject* tuple = PyTuple_New(2); PyObject* tuple = PyTuple_New(2);
PyTuple_SetItem(tuple, 0, a); PyTuple_SetItem(tuple, 0, a);
...@@ -5038,7 +5038,7 @@ CudaNdarray_Dimshuffle(PyObject* _unused, PyObject* args) ...@@ -5038,7 +5038,7 @@ CudaNdarray_Dimshuffle(PyObject* _unused, PyObject* args)
for (Py_ssize_t i = 0; i < pattern_dim; i++) for (Py_ssize_t i = 0; i < pattern_dim; i++)
{ {
PyObject * idx = PyLong_FromLong(i); PyObject * idx = PyLong_FromSsize_t(i);
if (idx == NULL) if (idx == NULL)
{ {
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论