Skip to content

Commit 05a9860

Browse files
committed
[MLIR Integreate] Updates for getBackwardSlice changes.
Signed-off-by: hanhanW <[email protected]>
1 parent 389541f commit 05a9860

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

lib/Dialect/Torch/Transforms/InlineGlobalSlots.cpp

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@
3232
#include "torch-mlir/Dialect/Torch/IR/TorchOps.h"
3333
#include "torch-mlir/Dialect/Torch/Transforms/Passes.h"
3434
#include "llvm/Support/Debug.h"
35+
#include "llvm/Support/LogicalResult.h"
3536

3637
#define DEBUG_TYPE "torch-inline-global-slots"
3738

@@ -251,7 +252,9 @@ bool InlineGlobalSlotsAnalysis::isValueSafeTransferFunction(Value value) {
251252

252253
SmallVector<Operation *> getBackwardSliceIncludingRoot(Value initialValue) {
253254
SetVector<Operation *> sliceSet;
254-
getBackwardSlice(initialValue, &sliceSet);
255+
[[maybe_unused]] LogicalResult ret =
256+
getBackwardSlice(initialValue, &sliceSet);
257+
assert(ret.succeeded());
255258
SmallVector<Operation *> slice;
256259
llvm::append_range(slice, sliceSet);
257260
slice.push_back(initialValue.getDefiningOp());

0 commit comments

Comments
 (0)