fix out of bounds array access in TilingParameters::bufferSizeOfLeaf()

tud-ccc · Sep 16, 2024 · f776a07 · f776a07
1 parent dcd57f8
commit f776a07
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/cinnamon/lib/Dialect/Cinm/Interfaces/TilingInterface.cpp b/cinnamon/lib/Dialect/Cinm/Interfaces/TilingInterface.cpp
@@ -118,9 +118,10 @@ int64_t TilingParameters::bufferSizeOfLeaf() {
   int64_t bufSize = 0;
   int i = 0;
   do {
-    int lastIdx = bufferSizesInBytes.size() - 1 - i;
+    size_t lastIdx = bufferSizesInBytes.size() - 1 - i;
     bufSize += bufferSizesInBytes[lastIdx] / numLeafsInDim;
-    numLeafsInDim *= workgroupShape[lastIdx];
+    numLeafsInDim *=
+        lastIdx < workgroupShape.size() ? workgroupShape[lastIdx] : 1;
     i++;
   } while (i < static_cast<int64_t>(bufferSizesInBytes.size()));
   return bufSize;