Skip to content

Commit

Permalink
[XLA:GPU][Emitters] Fix RUN directive in test.
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 675599571
  • Loading branch information
pifon2a authored and Google-ML-Automation committed Sep 18, 2024
1 parent c7c6d44 commit e22d2be
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 6 deletions.
4 changes: 2 additions & 2 deletions xla/service/gpu/fusions/transforms/tests/simplify_arith.mlir
Original file line number Diff line number Diff line change
@@ -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 {
Expand Down
6 changes: 2 additions & 4 deletions xla/service/gpu/model/indexing_map.cc
Original file line number Diff line number Diff line change
Expand Up @@ -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;
}

Expand Down

0 comments on commit e22d2be

Please sign in to comment.