Skip to content

Commit

Permalink
enable const prop in hybrid transform lit test
Browse files Browse the repository at this point in the history
Signed-off-by: Soren Lassen <[email protected]>
  • Loading branch information
sorenlassen committed Sep 19, 2023
1 parent 7aca52f commit c2bcf90
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
2 changes: 1 addition & 1 deletion src/Transform/ONNX/ConstProp.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -69,7 +69,7 @@ struct ConstPropONNXToONNXPassConfiguration {
};

int ConstPropONNXToONNXPassConfiguration::expansionBound = -1; // -1 == no bound
StringSet<> ConstPropONNXToONNXPassConfiguration::disabledPatterns;
StringSet<> ConstPropONNXToONNXPassConfiguration::disabledPatterns = {};
bool ConstPropONNXToONNXPassConfiguration::constantPropIsEnabled = false;

// Precondition: result has ranked tensor type with static shape and int or
Expand Down
4 changes: 2 additions & 2 deletions test/mlir/onnx/onnx_hybrid_transform.mlir
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
// RUN: onnx-mlir-opt -onnx-hybrid-transform=constant-propagation=false %s | FileCheck %s
// RUN: onnx-mlir-opt -onnx-hybrid-transform %s | FileCheck --check-prefix=CONSTPROP %s
// RUN: onnx-mlir-opt --enable-constant-prop=true -onnx-hybrid-transform=constant-propagation=false %s | FileCheck %s
// RUN: onnx-mlir-opt --enable-constant-prop=true -onnx-hybrid-transform %s | FileCheck --check-prefix=CONSTPROP %s

// Illustrates the back and forth between shape inference and the
// BinaryOpBroadcastAxisPattern canonicalization pattern:
Expand Down

0 comments on commit c2bcf90

Please sign in to comment.