[Mlir-commits] [mlir] 5110eb8 - [mlir][sparse] fix doc
Aart Bik
llvmlistbot at llvm.org
Tue Aug 9 17:19:46 PDT 2022
Author: Aart Bik
Date: 2022-08-09T17:19:39-07:00
New Revision: 5110eb8b677a779927c5ad8dc00c28eb0a41a0ae
URL: https://github.com/llvm/llvm-project/commit/5110eb8b677a779927c5ad8dc00c28eb0a41a0ae
DIFF: https://github.com/llvm/llvm-project/commit/5110eb8b677a779927c5ad8dc00c28eb0a41a0ae.diff
LOG: [mlir][sparse] fix doc
Reviewed By: Peiming
Differential Revision: https://reviews.llvm.org/D131527
Added:
Modified:
mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensorBase.td
mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensorOps.td
Removed:
################################################################################
diff --git a/mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensorBase.td b/mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensorBase.td
index d2186d954e288..2534bf1eb4e2a 100644
--- a/mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensorBase.td
+++ b/mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensorBase.td
@@ -59,7 +59,7 @@ def SparseTensor_Dialect : Dialect {
* [Biketal22] Aart J.C. Bik, Penporn Koanantakool, Tatiana Shpeisman,
Nicolas Vasilache, Bixia Zheng, and Fredrik Kjolstad. Compiler Support
for Sparse Tensor Computations in MLIR. ACM Transactions on Architecture
- and Code Optimization, June, 2022 (see https://dl.acm.org/doi/10.1145/3544559).
+ and Code Optimization, June, 2022. See: https://dl.acm.org/doi/10.1145/3544559
* [Chou18] Stephen Chou, Fredrik Berg Kjolstad, and Saman Amarasinghe.
Format Abstraction for Sparse Tensor Algebra Compilers. Proceedings of
the ACM on Programming Languages, October 2018.
diff --git a/mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensorOps.td b/mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensorOps.td
index f9200ed570d6c..39af4a846f247 100644
--- a/mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensorOps.td
+++ b/mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensorOps.td
@@ -514,6 +514,7 @@ def SparseTensor_UnaryOp : SparseTensor_Op<"unary", [NoSideEffect]>,
```
Example returning +1 for existing values and -1 for missing values:
+
```mlir
%result = sparse_tensor.unary %a : f64 to i32
present={
@@ -529,6 +530,7 @@ def SparseTensor_UnaryOp : SparseTensor_Op<"unary", [NoSideEffect]>,
Example showing a structural inversion (existing values become missing in
the output, while missing values are filled with 1):
+
```mlir
%result = sparse_tensor.unary %a : f64 to i64
present={}
@@ -562,7 +564,6 @@ def SparseTensor_ReduceOp : SparseTensor_Op<"reduce", [NoSideEffect, SameOperand
argument types.
Note that this operation is only required for custom reductions beyond the
-
standard operations (add, mul, and, or, etc). The `linalg.generic`
`iterator_types` defines which indices are being reduced. When the associated
operands are used in an operation, a reduction will occur. The use of this
More information about the Mlir-commits
mailing list