fix: encountered nullptr in RNN

Jittor · Apr 27, 2022 · 34740f9 · 34740f9
1 parent 0280f14
commit 34740f9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/python/jittor/__init__.py b/python/jittor/__init__.py
@@ -9,7 +9,7 @@
 # file 'LICENSE.txt', which is part of this source code package.
 # ***************************************************************
 
-__version__ = '1.3.3.12'
+__version__ = '1.3.3.13'
 from jittor_utils import lock
 with lock.lock_scope():
     ori_int = int

diff --git a/python/jittor/extern/cuda/cudnn/ops/cudnn_rnn_op.cc b/python/jittor/extern/cuda/cudnn/ops/cudnn_rnn_op.cc
@@ -115,6 +115,7 @@ void CudnnRnnOp::grads(Var** dout, VarPtr* dins) {
     VarPtr dy = dout[0];
     VarPtr dhy = dout[1];
     VarPtr dcy = cx ? dout[2] : nullptr;
+    if (!dy.ptr) dy = make_number(0.0, y);
     if (!dhy.ptr) dhy = make_number(0.0, hy);
     if (!dcy.ptr && cx) dcy = make_number(0.0, cy);