diff --git a/tests/common/src/dot.h b/tests/common/src/dot.h index 7511eda9a..8521cd0a6 100644 --- a/tests/common/src/dot.h +++ b/tests/common/src/dot.h @@ -64,7 +64,7 @@ void dot(TYPE * A, TYPE * B, TYPE * C, // epilogue int rxc[TM] = ridx * TM + 0 ... TM; int ryc[TN] = ridy * TN + 0 ... TN; - TYPE* pc[TM, TN] = C + ryc[newaxis, :] + rxc[:, newaxis] * ldc; + TYPE* pc[TM, TN] = C + rxc[:, newaxis] * ldc + ryc[newaxis, :]; *pc = c; } )";