diff --git a/mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensorOps.td b/mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensorOps.td --- a/mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensorOps.td +++ b/mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensorOps.td @@ -629,7 +629,7 @@ // sparse tensor codegen to progressively lower sparse tensors. //===----------------------------------------------------------------------===// -def SparseTensor_StorageNewOp : SparseTensor_Op<"storage", []>, +def SparseTensor_StorageOp : SparseTensor_Op<"storage", []>, Arguments<(ins Variadic:$inputs)>, Results<(outs AnyTuple:$result)> { let summary = "Pack a list of value into one sparse tensor storage value"; diff --git a/mlir/lib/Dialect/SparseTensor/IR/SparseTensorDialect.cpp b/mlir/lib/Dialect/SparseTensor/IR/SparseTensorDialect.cpp --- a/mlir/lib/Dialect/SparseTensor/IR/SparseTensorDialect.cpp +++ b/mlir/lib/Dialect/SparseTensor/IR/SparseTensorDialect.cpp @@ -486,7 +486,7 @@ // Sparse Tensor Storage Operation. //===----------------------------------------------------------------------===// -LogicalResult StorageNewOp::verify() { +LogicalResult StorageOp::verify() { auto retTypes = getResult().getType().getTypes(); if (retTypes.size() != getInputs().size()) return emitError("The number of inputs is inconsistent with output tuple"); diff --git a/mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorCodegen.cpp b/mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorCodegen.cpp --- a/mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorCodegen.cpp +++ b/mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorCodegen.cpp @@ -174,7 +174,7 @@ /// Creates tuple. static Value createTupleMake(OpBuilder &builder, Location loc, Type type, ValueRange values) { - return builder.create(loc, type, values); + return builder.create(loc, type, values); } /// Create allocation operation. diff --git a/mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorPasses.cpp b/mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorPasses.cpp --- a/mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorPasses.cpp +++ b/mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorPasses.cpp @@ -156,7 +156,7 @@ ConversionTarget target(*ctx); // Almost everything in the sparse dialect must go! target.addIllegalDialect(); - target.addLegalOp(); + target.addLegalOp(); // All dynamic rules below accept new function, call, return, and various // tensor and bufferization operations as legal output of the rewriting // provided that all sparse tensor types have been fully rewritten.