提交 11df7d41 authored 作者: abergeron's avatar abergeron

Merge pull request #2830 from nouiz/crash_curand_mac

Fix crash on Mac with curand op. We where using g++ extention that MSVC ...
......@@ -167,8 +167,7 @@ class CURAND_Base(GpuOp):
}
for (int i = 0; i < %(ndim)s; ++i)
{
odims[i] = ((npy_int32*)(PyArray_DATA(%(size)s) +
PyArray_STRIDES(%(size)s)[0] * i))[0];
odims[i] = ((npy_int32*)PyArray_GETPTR1(%(size)s, i))[0];
n_elements *= odims[i];
must_alloc_sample = (must_alloc_sample
|| CudaNdarray_HOST_DIMS(%(o_sample)s)[i] != odims[i]);
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论