[Mlir-commits] [mlir] [mlir][sparse][nfc] fixed typo in "translate" (PR #83891)
Aart Bik
llvmlistbot at llvm.org
Mon Mar 4 10:48:24 PST 2024
https://github.com/aartbik created https://github.com/llvm/llvm-project/pull/83891
None
>From 14de28a823f5e98e1d85d726a878f0e048177e54 Mon Sep 17 00:00:00 2001
From: Aart Bik <ajcbik at google.com>
Date: Mon, 4 Mar 2024 10:22:09 -0800
Subject: [PATCH] [mlir][sparse][nfc] fixed typo in "translate"
---
.../mlir/Dialect/SparseTensor/IR/SparseTensorAttrDefs.td | 2 +-
.../mlir/Dialect/SparseTensor/IR/SparseTensorType.h | 8 ++++----
mlir/lib/Dialect/SparseTensor/IR/SparseTensorDialect.cpp | 6 +++---
3 files changed, 8 insertions(+), 8 deletions(-)
diff --git a/mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensorAttrDefs.td b/mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensorAttrDefs.td
index 5d1db2323f95f0..d3be8a3009ba1e 100644
--- a/mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensorAttrDefs.td
+++ b/mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensorAttrDefs.td
@@ -431,7 +431,7 @@ def SparseTensorEncodingAttr : SparseTensor_Attr<"SparseTensorEncoding",
//
// Helper function to translate between level/dimension space.
//
- SmallVector<int64_t> tranlateShape(::mlir::ArrayRef<int64_t> srcShape, ::mlir::sparse_tensor::CrdTransDirectionKind) const;
+ SmallVector<int64_t> translateShape(::mlir::ArrayRef<int64_t> srcShape, ::mlir::sparse_tensor::CrdTransDirectionKind) const;
ValueRange translateCrds(::mlir::OpBuilder &builder, ::mlir::Location loc, ::mlir::ValueRange crds, ::mlir::sparse_tensor::CrdTransDirectionKind) const;
//
diff --git a/mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensorType.h b/mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensorType.h
index c93a4fcd922c28..bd2c3c1dd55159 100644
--- a/mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensorType.h
+++ b/mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensorType.h
@@ -249,14 +249,14 @@ class SparseTensorType {
/// Returns the Level-shape.
SmallVector<Size> getLvlShape() const {
- return getEncoding().tranlateShape(getDimShape(),
- CrdTransDirectionKind::dim2lvl);
+ return getEncoding().translateShape(getDimShape(),
+ CrdTransDirectionKind::dim2lvl);
}
/// Returns the Level-shape.
SmallVector<Size> getBatchLvlShape() const {
- auto lvlShape = getEncoding().tranlateShape(getDimShape(),
- CrdTransDirectionKind::dim2lvl);
+ auto lvlShape = getEncoding().translateShape(
+ getDimShape(), CrdTransDirectionKind::dim2lvl);
lvlShape.truncate(getEncoding().getBatchLvlRank());
return lvlShape;
}
diff --git a/mlir/lib/Dialect/SparseTensor/IR/SparseTensorDialect.cpp b/mlir/lib/Dialect/SparseTensor/IR/SparseTensorDialect.cpp
index 232635ca84a47e..244a082d04870e 100644
--- a/mlir/lib/Dialect/SparseTensor/IR/SparseTensorDialect.cpp
+++ b/mlir/lib/Dialect/SparseTensor/IR/SparseTensorDialect.cpp
@@ -413,8 +413,8 @@ SparseTensorEncodingAttr::getStaticLvlSliceStride(Level lvl) const {
}
SmallVector<int64_t>
-SparseTensorEncodingAttr::tranlateShape(ArrayRef<int64_t> srcShape,
- CrdTransDirectionKind dir) const {
+SparseTensorEncodingAttr::translateShape(ArrayRef<int64_t> srcShape,
+ CrdTransDirectionKind dir) const {
if (isIdentity())
return SmallVector<int64_t>(srcShape);
@@ -1392,7 +1392,7 @@ void ReinterpretMapOp::build(OpBuilder &odsBuilder, OperationState &odsState,
auto srcStt = getSparseTensorType(source);
SmallVector<int64_t> srcLvlShape = srcStt.getLvlShape();
SmallVector<int64_t> dstDimShape =
- dstEnc.tranlateShape(srcLvlShape, CrdTransDirectionKind::lvl2dim);
+ dstEnc.translateShape(srcLvlShape, CrdTransDirectionKind::lvl2dim);
auto dstTp =
RankedTensorType::get(dstDimShape, srcStt.getElementType(), dstEnc);
return build(odsBuilder, odsState, dstTp, source);
More information about the Mlir-commits
mailing list