Pass indexing map by reference

PiperOrigin-RevId: 659922479
openxla · Aug 6, 2024 · 6eef582 · 6eef582
1 parent 4756a86
commit 6eef582
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/xla/service/gpu/fusions/mlir/ir/xla_gpu_ops.cc b/xla/service/gpu/fusions/mlir/ir/xla_gpu_ops.cc
@@ -123,7 +123,8 @@ void ApplyIndexingOp::build(OpBuilder& builder, OperationState& result,
 }
 
 void ApplyIndexingOp::build(OpBuilder& builder, OperationState& result,
-                            ValueRange operands, IndexingMap indexing_map) {
+                            ValueRange operands,
+                            const IndexingMap& indexing_map) {
   SmallVector<Type, 2> result_types(indexing_map.GetAffineMap().getNumResults(),
                                     builder.getIndexType());
   IndexingMapAttr indexing_map_attr =

diff --git a/xla/service/gpu/fusions/mlir/ir/xla_gpu_ops.td b/xla/service/gpu/fusions/mlir/ir/xla_gpu_ops.td
@@ -253,7 +253,7 @@ def ApplyIndexingOp : XLAGPU_Op<"apply_indexing", [Pure]> {
     OpBuilder<(ins "mlir::ValueRange":$dims, "mlir::ValueRange":$symbols,
                    "const IndexingMap&":$indexing_map)>,
     OpBuilder<(ins "mlir::ValueRange":$operands,
-                   "IndexingMap":$indexing_map)>,
+                   "const IndexingMap&":$indexing_map)>,
     OpBuilder<(ins "mlir::ValueRange":$operands, "mlir::AffineMap":$affine_map,
                "llvm::ArrayRef<DimVar>":$dim_vars,
                "llvm::ArrayRef<RangeVar>":$range_vars)>,