[mlir][sparse] Fix a problem in the new operator rewriter.

The getSparseTensorReaderNextX functions should return void.

Reviewed By: aartbik

Differential Revision: https://reviews.llvm.org/D138226
This commit is contained in:
bixia1
2022-11-17 10:19:57 -08:00
parent 534b26aa07
commit 96b3bf4292
2 changed files with 15 additions and 9 deletions

View File

@@ -930,9 +930,8 @@ struct NewRewriter : public OpRewritePattern<NewOp> {
SmallString<18> getNextFuncName{"getSparseTensorReaderNext",
primaryTypeFunctionSuffix(eltTp)};
Value indices = dimSizes; // Reuse the indices memref to store indices.
createFuncCall(rewriter, loc, getNextFuncName, {eltTp},
{reader, indices, value}, EmitCInterface::On)
.getResult(0);
createFuncCall(rewriter, loc, getNextFuncName, {}, {reader, indices, value},
EmitCInterface::On);
SmallVector<Value> indicesArray;
for (uint64_t i = 0; i < rank; i++) {
indicesArray.push_back(rewriter.create<memref::LoadOp>(