提交 ce0494c9 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Fix segfault.

上级 3e4b1d77
...@@ -193,7 +193,7 @@ int dnn_rnn_gi(cudnnRNNDescriptor_t desc, npy_uint64 xshp, ...@@ -193,7 +193,7 @@ int dnn_rnn_gi(cudnnRNNDescriptor_t desc, npy_uint64 xshp,
yl, PyGpuArray_DEV_DATA(y), yl, PyGpuArray_DEV_DATA(y),
/* y and dy are the same shape */ /* y and dy are the same shape */
yl, PyGpuArray_DEV_DATA(dy), yl, PyGpuArray_DEV_DATA(dy),
dhydesc, PyGpuArray_DEV_DATA(dhy), dhydesc, dhy ? PyGpuArray_DEV_DATA(dhy) : NULL,
dcydesc, dcy ? PyGpuArray_DEV_DATA(dcy) : NULL, dcydesc, dcy ? PyGpuArray_DEV_DATA(dcy) : NULL,
wdesc, PyGpuArray_DEV_DATA(w), wdesc, PyGpuArray_DEV_DATA(w),
hxdesc, PyGpuArray_DEV_DATA(hx), hxdesc, PyGpuArray_DEV_DATA(hx),
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论