[XLA:GPU][Emitters] Fix RUN directive in test.

PiperOrigin-RevId: 675862232
openxla · Sep 18, 2024 · 5b0dfe3 · 5b0dfe3
1 parent c7c6d44
commit 5b0dfe3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/xla/service/gpu/fusions/transforms/tests/simplify_arith.mlir b/xla/service/gpu/fusions/transforms/tests/simplify_arith.mlir
@@ -1,5 +1,5 @@
-// RUN: mlir_fusions_opt %s -split-input-file -xla-gpu-simplify-arith -cse -canonicalize  --mlir-print-ir-after-all
-//| FileCheck %s
+// RUN: mlir_fusions_opt %s -split-input-file -xla-gpu-simplify-arith -cse \
+// RUN:   -canonicalize | FileCheck %s
 
 module {
   func.func @unknown(%arg0: index {xla.range = [0 : index, 42 : index]}) -> i1 {

diff --git a/xla/service/gpu/model/indexing_map.cc b/xla/service/gpu/model/indexing_map.cc
@@ -2172,10 +2172,8 @@ IndexingMap IndexingMap::ConvertSymbolsToDimensions() const {
   AffineMap canonical_map =
       affine_map_.replaceDimsAndSymbols({}, syms_replacements, num_vars, 0);
   IndexingMap new_indexing_map(canonical_map, new_dim_vars, /*range_vars=*/{},
-                               /*rt_vars=*/{}, new_constraints, is_simplified_);
-  if (is_simplified_) {
-    new_indexing_map.Simplify();
-  }
+                               /*rt_vars=*/{}, new_constraints,
+                               /*is_simplified=*/false);
   return new_indexing_map;
 }