[mlir][sparse] Fix Python interface for bufferization.alloc_tensor.
authorbixia1 <bixia@google.com>
Thu, 10 Nov 2022 01:33:25 +0000 (17:33 -0800)
committerbixia1 <bixia@google.com>
Thu, 10 Nov 2022 05:38:51 +0000 (21:38 -0800)
commitf151274454c2b55c5c64363ddc590d489d26c3fc
tree6f9403a511802c897366dc3f6d0a8ecf2082a9a5
parent8993f3eed64bd1592a8478d0a6a608e64f8edc01
[mlir][sparse] Fix Python interface for bufferization.alloc_tensor.

Add size_hint operand to the Python interface.
Fix pytaco.

Reviewed By: aartbik

Differential Revision: https://reviews.llvm.org/D137754
mlir/python/mlir/dialects/_bufferization_ops_ext.py
mlir/test/Integration/Dialect/SparseTensor/taco/tools/mlir_pytaco.py