Erase InsertSliceOp when converting to stream

UIUC-ChenLab · Feb 20, 2024 · efd7f0e · efd7f0e
1 parent 3f9391d
commit efd7f0e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/lib/Dialect/HLS/TransformOps/HLSTransformOps.cpp b/lib/Dialect/HLS/TransformOps/HLSTransformOps.cpp
@@ -201,6 +201,7 @@ transform::HLSConvertInsertSliceToStreamOp::applyToOne(
       rewriter.getUnknownLoc(), tensorToReplace.getType(), channel);
 
   rewriter.replaceAllUsesWith(tensorToReplace, channelTensor);
+  rewriter.replaceOp(insertSlice, insertSlice.getDest());
   results.push_back(channel);
   results.push_back(channelWrite);
   results.push_back(channelTensor);