[mlir][sparse] fix bug when packing tensor with 32 bit pointer width.

Reviewed By: wrengr Differential Revision: https://reviews.llvm.org/D143450
2023-02-07 01:21:10 +00:00
parent 1d1a2eb298
commit 7a8edea69d
2 changed files with 38 additions and 2 deletions
--- a/mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorCodegen.cpp
+++ b/mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorCodegen.cpp
@@ -1060,8 +1060,11 @@ struct SparsePackOpConverter : public OpConversionPattern<PackOp> {
                loc, tensorType,
                DenseElementsAttr::get(
                    tensorType,
-                    {APInt(64, 0),
-                     APInt(64, op.getData().getType().getShape()[0])}));
+                    ArrayRef<Attribute>{
+                        IntegerAttr::get(enc.getPointerType(), 0),
+                        IntegerAttr::get(
+                            enc.getPointerType(),
+                            op.getData().getType().getShape()[0])}));
            field = rewriter.create<bufferization::ToMemrefOp>(loc, memrefType,
                                                               cstPtr);
            break;