benchdnn: rnn: fix in dst_layer dereferencing for bidir_concat

oneapi-src · Aug 12, 2019 · cf83e83 · cf83e83
1 parent 1188010
commit cf83e83
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/benchdnn/rnn/rnn.hpp b/tests/benchdnn/rnn/rnn.hpp
@@ -406,8 +406,9 @@ inline size_t tnc_off_f(const prb_t *p, int64_t t, int64_t n, int64_t c) {
 
 inline void inv_tnc_off_f(
         const prb_t *p, size_t off, int64_t &t, int64_t &n, int64_t &c) {
-    c = off % p->sic;
-    off /= p->sic;
+    auto cout = p->sic * (1 + (p->direction == mkldnn_bidirectional_concat));
+    c = off % cout;
+    off /= cout;
     n = off % p->mb;
     off /= p->mb;
     t = off % p->n_iter;