Fix vector C offset computation logic.

llvm · Sep 18, 2020 · d9bd733 · d9bd733
1 parent 2467036
commit d9bd733
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mlir/include/mlir/Dialect/MIOpen/LowerMIOpenOps.h b/mlir/include/mlir/Dialect/MIOpen/LowerMIOpenOps.h
@@ -3433,7 +3433,7 @@ struct GridwiseGemmV2RewritePattern : public OpRewritePattern<miopen::GridwiseGe
     int64_t NumBlks = NumBlksPerXdlops * MRepeats * NRepeats;
 
     int64_t iterationsPerVectorC = NumBlks / vectorNumber;
-    int64_t vectorCoffset = vectorType.getShape()[0] / BlkSize;
+    int64_t vectorCoffset = vectorType.getShape()[0] / iterationsPerVectorC;
 
     // llvm::errs() << "MPerWave: " << MPerWave << "\n";
     // llvm::errs() << "NPerWave: " << NPerWave << "\n\n";