Fix getBuffersize() for CategoryMapperOp

Signed-off-by: Haruki Imai <imaihal@jp.ibm.com>
imaihal · Aug 30, 2024 · b99a334 · b99a334
1 parent 18b9919
commit b99a334
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/Dialect/Krnl/KrnlOps.cpp b/src/Dialect/Krnl/KrnlOps.cpp
@@ -840,7 +840,8 @@ uint64_t KrnlGlobalOp::getBufferSize() {
   KrnlGlobalOp krnlGlobalOp = mlir::cast<KrnlGlobalOp>(getOperation());
   const Type type = krnlGlobalOp.getResult().getType();
   const MemRefType memRefTy = mlir::cast<mlir::MemRefType>(type);
-  return affine::getIntOrFloatMemRefSizeInBytes(memRefTy).value();
+  auto sizeInBytes = affine::getIntOrFloatMemRefSizeInBytes(memRefTy);
+  return sizeInBytes.has_value() ? sizeInBytes.value() : 0;
 }
 
 void KrnlGlobalOp::setBuffer(ArrayRef<char> rawData) { return; }