Skip to content

Commit

Permalink
Revert options formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
melissawm committed Dec 19, 2024
1 parent 5c8284d commit a953af0
Showing 1 changed file with 9 additions and 2 deletions.
11 changes: 9 additions & 2 deletions shardy/dialect/sdy/transforms/propagation/passes.td
Original file line number Diff line number Diff line change
Expand Up @@ -149,6 +149,13 @@ def PopulateOpShardingRulesPass : Pass<"sdy-populate-op-sharding-rules", "func::
sharding axes during propagation.
}];
let dependentDialects = ["mlir::sdy::SdyDialect"];
// TODO(b/347180954): remove conservative propagation once the cost model
// supports split axes and padding.

let options = [
// TODO(b/347180954): remove conservative propagation once the cost model
// supports split axes and padding.
Option<"conservativePropagation", "conservative-propagation", "bool",
/*default=*/"false",
"whether to disllow rules that can propagate non-divisible sharding "
"axes">
];
}

0 comments on commit a953af0

Please sign in to comment.