From 1befa4f408ba1fd489b5a53a9e682da2ec16fa3a Mon Sep 17 00:00:00 2001 From: Peter Buchlovsky Date: Wed, 6 Aug 2025 03:53:22 -0700 Subject: [PATCH] [mpmd] Test different order PiperOrigin-RevId: 791620029 --- .../dialect/mpmd/transforms/common/test/rule_based_merge.mlir | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/shardy/dialect/mpmd/transforms/common/test/rule_based_merge.mlir b/shardy/dialect/mpmd/transforms/common/test/rule_based_merge.mlir index 48f7afb0..a9f9003a 100644 --- a/shardy/dialect/mpmd/transforms/common/test/rule_based_merge.mlir +++ b/shardy/dialect/mpmd/transforms/common/test/rule_based_merge.mlir @@ -1,4 +1,4 @@ -// RUN: mpmd_opt %s -mpmd-rule-based-merge='rules=FragmentMergeRule(sources=[FragmentInfo(origins=["f"]),FragmentInfo(origins=["g"])],target=FragmentInfo(origins=["f","g"])),FragmentMergeRule(sources=[FragmentInfo(origins=["i"]),FragmentInfo(origins=["j"]),FragmentInfo(origins=["k"])],target=FragmentInfo(origins=["i","j","k"]))'2>&1 | FileCheck %s +// RUN: mpmd_opt %s -mpmd-rule-based-merge='rules=FragmentMergeRule(sources=[FragmentInfo(origins=["f"]),FragmentInfo(origins=["g"])],target=FragmentInfo(origins=["f","g"])),FragmentMergeRule(sources=[FragmentInfo(origins=["j"]),FragmentInfo(origins=["i"]),FragmentInfo(origins=["k"])],target=FragmentInfo(origins=["i","j","k"]))'2>&1 | FileCheck %s // TODO(b/435182733) Add more lit tests for rule based merge pass. !mesh_1_tensor_2_2_f32 = !mpmd.mesh_tensor<"m1", tensor<2x2xf32>>