Skip to content

Commit fd3b51f

Browse files
committed
Fix cache issue with strides
1 parent 3bc8548 commit fd3b51f

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

include/matx/transforms/matmul.h

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -997,7 +997,9 @@ struct MatMulParamsKeyEq {
997997
return l.m == t.m && l.n == t.n && l.k == t.k && l.a_rows == t.a_rows &&
998998
l.b_rows == t.b_rows && l.c_rows == t.c_rows &&
999999
l.a_cols == t.a_cols && l.b_cols == t.b_cols &&
1000-
l.c_cols == t.c_cols && l.stream == t.stream && l.lda == t.lda &&
1000+
l.c_cols == t.c_cols &&
1001+
l.astride == t.astride && l.bstride == t.bstride && l.cstride == t.cstride &&
1002+
l.stream == t.stream && l.lda == t.lda &&
10011003
l.ldb == t.ldb && l.ldc == t.ldc && l.batch == t.batch &&
10021004
l.prov == t.prov && l.dtype == t.dtype && l.opA == t.opA &&
10031005
l.opB == t.opB;

0 commit comments

Comments
 (0)