caio/clang-p2996
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[mlir][sparse] support non-id map for [Dis]assembleOp (#80355)

Browse Source
This commit is contained in:
Peiming Liu
2024-02-01 17:11:33 -06:00
committed by GitHub
parent 375bd2201c
commit 07bf1ddb4e
3 changed files with 84 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
mlir/lib/Dialect/SparseTensor/IR/SparseTensorDialect.cpp
View File

@@ -1016,8 +1016,6 @@ static LogicalResult verifyPackUnPack(Operation *op, bool requiresStaticShape,
return op->emitError("the sparse-tensor must have static shape");
if (!stt.hasEncoding())
return op->emitError("the sparse-tensor must have an encoding attribute");
if (!stt.isIdentity())
return op->emitError("the sparse-tensor must have the identity mapping");
// Verifies the trailing COO.
Level cooStartLvl = stt.getCOOStart();