提交 a33c5a18 authored 作者: Frederic's avatar Frederic

more clear code

上级 f9aa9f5e
......@@ -167,8 +167,7 @@ class CURAND_Base(GpuOp):
}
for (int i = 0; i < %(ndim)s; ++i)
{
odims[i] = ((npy_int32*)(PyArray_BYTES(%(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 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论