Fix a bug when removing duplicated code.

Signed-off-by: Haruki Imai <imaihal@jp.ibm.com>
onnx · Oct 10, 2024 · 1a4cb5d · 1a4cb5d
1 parent 92e831b
commit 1a4cb5d
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/src/Dialect/Krnl/KrnlOps.cpp b/src/Dialect/Krnl/KrnlOps.cpp
@@ -815,16 +815,18 @@ MutableOperandRange KrnlSpecializedKernel::getLoopRefs() {
 }
 
 ArrayRef<char> KrnlGlobalOp::getBuffer() {
-  ArrayRef<char> rawData;
+  ArrayRef<char> ret;
   std::vector<char> attrData;
   auto krnlGlobalOp = mlir::cast<KrnlGlobalOp>(*getOperation());
   if (krnlGlobalOp.getValueAttr()) {
+    int64_t sizeInBytes = getBufferSize();
+    char *rawData = (char *)malloc(sizeInBytes);
     auto valueAttr = krnlGlobalOp.getValue().value();
     getRawData(valueAttr, attrData);
-    int64_t sizeInBytes = getBufferSize();
-    rawData = llvm::ArrayRef(attrData.data(), sizeInBytes);
+    memcpy(rawData, attrData.data(), sizeInBytes);
+    ret = llvm::ArrayRef(rawData, sizeInBytes);
   }
-  return rawData;
+  return ret;
 }
 
 uint64_t KrnlGlobalOp::getBufferSize() {