From a953af0be4741b858ac45771580d9be221a7d009 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Melissa=20Weber=20Mendon=C3=A7a?= Date: Thu, 19 Dec 2024 16:12:29 -0300 Subject: [PATCH] Revert options formatting --- shardy/dialect/sdy/transforms/propagation/passes.td | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/shardy/dialect/sdy/transforms/propagation/passes.td b/shardy/dialect/sdy/transforms/propagation/passes.td index 3d6a73e..2cae54b 100644 --- a/shardy/dialect/sdy/transforms/propagation/passes.td +++ b/shardy/dialect/sdy/transforms/propagation/passes.td @@ -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"> + ]; }