Matthias Springer 77f8297c6f [mlir][sparse] Improve sparse tensor type constraints (#112133) ...

Sparse tensors are always ranked tensors. Encodings cannot be attached
to unranked tensors. Change the type constraint to `RankedTensorOf`, so
that we generate `TypedValue<RankedTensorType>` instead of
`TypedValue<TensorType>`. This removes the need for type casting in some
cases.

Also improve the verifiers (missing `return` statements) and switch a
few other `AnyTensor` to `AnyRankedTensor`.

This commit is in preparation of a dialect conversion commit that
required fixes in the sparse dialect.

2024-10-13 21:12:38 +02:00

..

GPU

[mlir][sparse] use a consistent order between [dis]assembleOp and sto… (#84079)

2024-03-06 09:57:41 -08:00

binary_valued.mlir

[mlir][sparse] force a properly sized view on pos/crd/val under codegen (#91288)

2024-05-07 09:20:56 -07:00

buffer_rewriting.mlir

…

canonicalize.mlir

[mlir][sparse] add canonicalization patterns for IterateOp. (#95569)

2024-06-14 10:31:29 -07:00

codegen_buffer_initialization.mlir

…

codegen_sparse_alloc.mlir

…

codegen_to_llvm.mlir

…

codegen.mlir

[mlir][sparse][test] Adjust tests for LowerSparseOpsToForeach (#110976)

2024-10-11 15:56:39 +08:00

constant_index_map.mlir

[mlir][sparse] reuse tensor.insert operation to insert elements into … (#84987)

2024-03-12 16:59:17 -07:00

conversion.mlir

[mlir][llvm] Implement ConstantLike for ZeroOp, UndefOp, PoisonOp (#93690)

2024-05-30 08:21:08 +02:00

convert_dense2sparse.mlir

[mlir][sparse] deallocate tmp coo buffer generated during stage-spars… (#82017)

2024-02-17 12:17:57 -08:00

convert_sparse2dense.mlir

…

convert_sparse2sparse_element.mlir

…

convert_sparse2sparse.mlir

…

dense.mlir

…

external_direct.mlir

[mlir][sparse] allow for direct-out passing of sparse tensor buffers (#88327)

2024-04-11 10:07:24 -07:00

external.mlir

[mlir][sparse] use a consistent order between [dis]assembleOp and sto… (#84079)

2024-03-06 09:57:41 -08:00

fold.mlir

…

fuse_sparse_concat_with_extract_slice.mlir

[mlir][sparse] fuse concat and extract_slice op if possible. (#89825)

2024-04-24 13:51:41 -07:00

fuse_sparse_convert_into_producer.mlir

[mlir][sparse] clone a empty sparse tensor when fuse convert into pro… (#92158)

2024-05-14 13:26:49 -07:00

fuse_sparse_pad_with_consumer.mlir

[mlir][sparse] support sparsifying sparse kernels to sparse-iterator-based loop (#95858)

2024-06-17 16:50:12 -07:00

invalid_encoding.mlir

[mlir][sparse] Add verification for explicit/implicit value (#90111)

2024-05-07 20:28:39 -04:00

invalid.mlir

[mlir][sparse] Improve sparse tensor type constraints (#112133)

2024-10-13 21:12:38 +02:00

minipipeline_parallel.mlir

[mlir][sparse] add parallelization options to mini pipeline (#104233)

2024-08-30 17:47:37 -04:00

minipipeline_vector.mlir

[mlir][sparse] expose optimization flags to mini pipeline (#95158)

2024-06-11 14:20:58 -07:00

no_lowering.mlir

[mlir][sparse][test] Adjust tests for LowerSparseOpsToForeach (#110976)

2024-10-11 15:56:39 +08:00

one_shot_bufferize_invalid.mlir

…

one_shot_bufferize_tensor_copy_insertion.mlir

[mlir][linalg][bufferize] Fix element-wise access optimization for sparse tensors (#87305)

2024-04-03 09:57:25 +09:00

one_trip.mlir

…

pack_copy.mlir

[mlir][sparse] use a consistent order between [dis]assembleOp and sto… (#84079)

2024-03-06 09:57:41 -08:00

pre_rewriting.mlir

…

rejected.mlir

…

rewriting_for_codegen.mlir

…

roundtrip_encoding.mlir

[mlir][sparse] allow multiple COO segments in sparse encodings. (#91786)

2024-05-10 11:36:01 -07:00

roundtrip.mlir

[mlir][sparse] introduce sparse_tensor.coiterate operation. (#101100)

2024-07-31 15:14:27 -07:00

scf_1_N_conversion.mlir

…

semi_ring.mlir

…

sorted_coo.mlir

…

sparse_1d.mlir

…

sparse_2d.mlir

[mlir][sparse] support sparsifying sparse kernels to sparse-iterator-based loop (#95858)

2024-06-17 16:50:12 -07:00

sparse_3d.mlir

[mlir][sparse] support sparsifying sparse kernels to sparse-iterator-based loop (#95858)

2024-06-17 16:50:12 -07:00

sparse_affine.mlir

[mlir][sparse] support sparsifying sparse kernels to sparse-iterator-based loop (#95858)

2024-06-17 16:50:12 -07:00

sparse_batch.mlir

[mlir][sparse] support sparsifying sparse kernels to sparse-iterator-based loop (#95858)

2024-06-17 16:50:12 -07:00

sparse_broadcast.mlir

[mlir][sparse] support sparsifying sparse kernels to sparse-iterator-based loop (#95858)

2024-06-17 16:50:12 -07:00

sparse_concat.mlir

[mlir][sparse] support sparsifying sparse kernels to sparse-iterator-based loop (#95858)

2024-06-17 16:50:12 -07:00

sparse_conv_2d_slice_based.mlir

[mlir][sparse] reuse tensor.insert operation to insert elements into … (#84987)

2024-03-12 16:59:17 -07:00

sparse_expand.mlir

…

sparse_extract_slice.mlir

…

sparse_fill_zero.mlir

[mlir][sparse] support sparsifying sparse kernels to sparse-iterator-based loop (#95858)

2024-06-17 16:50:12 -07:00

sparse_foreach.mlir

…

sparse_fp_ops.mlir

[mlir][sparse] support sparsifying sparse kernels to sparse-iterator-based loop (#95858)

2024-06-17 16:50:12 -07:00

sparse_fusion.mlir

[mlir][sparse] support sparsifying sparse kernels to sparse-iterator-based loop (#95858)

2024-06-17 16:50:12 -07:00

sparse_index.mlir

[mlir][sparse] reuse tensor.insert operation to insert elements into … (#84987)

2024-03-12 16:59:17 -07:00

sparse_int_ops.mlir

[mlir][sparse] support sparsifying sparse kernels to sparse-iterator-based loop (#95858)

2024-06-17 16:50:12 -07:00

sparse_iteration_to_scf.mlir

[mlir][sparse] expose emit strategy option to mini pipeline (#96238)

2024-06-20 14:23:36 -07:00

sparse_itertion_licm.mlir

[mlir][sparse] introduce sparse_tensor.iterate operation (#88955)

2024-06-10 10:20:24 -07:00

sparse_kernels_to_iterator.mlir

[mlir][sparse] partially support lowering sparse coiteration loops to scf.while/for. (#105565)

2024-08-23 10:47:44 -07:00

sparse_kernels.mlir

[mlir][sparse] support sparsifying sparse kernels to sparse-iterator-based loop (#95858)

2024-06-17 16:50:12 -07:00

sparse_lower_col.mlir

[mlir] Remove dialect specific bufferization passes (Reland) (#93535)

2024-05-28 20:04:27 +01:00

sparse_lower_inplace.mlir

[mlir][sparse] support sparsifying sparse kernels to sparse-iterator-based loop (#95858)

2024-06-17 16:50:12 -07:00

sparse_lower.mlir

[mlir] Remove dialect specific bufferization passes (Reland) (#93535)

2024-05-28 20:04:27 +01:00

sparse_matmul_codegen.mlir

[mlir][sparse] support sparsifying sparse kernels to sparse-iterator-based loop (#95858)

2024-06-17 16:50:12 -07:00

sparse_matmul_one.mlir

[mlir][sparse] Support explicit/implicit value for complex type (#90771)

2024-05-02 12:28:34 -04:00

sparse_nd.mlir

[mlir][sparse] support sparsifying sparse kernels to sparse-iterator-based loop (#95858)

2024-06-17 16:50:12 -07:00

sparse_out.mlir

[mlir][sparse] support sparsifying sparse kernels to sparse-iterator-based loop (#95858)

2024-06-17 16:50:12 -07:00

sparse_outbuf.mlir

[mlir][sparse] support sparsifying sparse kernels to sparse-iterator-based loop (#95858)

2024-06-17 16:50:12 -07:00

sparse_pack.mlir

[mlir][sparse] use a consistent order between [dis]assembleOp and sto… (#84079)

2024-03-06 09:57:41 -08:00

sparse_parallel_reduce.mlir

[mlir][sparse] support sparsifying sparse kernels to sparse-iterator-based loop (#95858)

2024-06-17 16:50:12 -07:00

sparse_parallel.mlir

…

sparse_perm_lower.mlir

…

sparse_perm.mlir

[mlir][sparse] support sparsifying sparse kernels to sparse-iterator-based loop (#95858)

2024-06-17 16:50:12 -07:00

sparse_reinterpret_map.mlir

[mlir][sparse] reuse tensor.insert operation to insert elements into … (#84987)

2024-03-12 16:59:17 -07:00

sparse_relu.mlir

[mlir][sparse] recognize ReLu operation during sparsification (#92016)

2024-05-13 14:02:29 -07:00

sparse_reshape.mlir

[mlir][sparse] support sparsifying sparse kernels to sparse-iterator-based loop (#95858)

2024-06-17 16:50:12 -07:00

sparse_scalars.mlir

…

sparse_sddmm_org.mlir

[mlir][sparse] support sparsifying sparse kernels to sparse-iterator-based loop (#95858)

2024-06-17 16:50:12 -07:00

sparse_sddmm.mlir

[mlir][sparse] support sparsifying sparse kernels to sparse-iterator-based loop (#95858)

2024-06-17 16:50:12 -07:00

sparse_space_collapse.mlir

[mlir][sparse] implement lowering rules for ExtractIterSpaceOp. (#89143)

2024-06-12 10:49:12 -07:00

sparse_storage.mlir

…

sparse_tensor_reshape.mlir

[mlir][sparse] support sparsifying sparse kernels to sparse-iterator-based loop (#95858)

2024-06-17 16:50:12 -07:00

sparse_transpose.mlir

[mlir][sparse] reuse tensor.insert operation to insert elements into … (#84987)

2024-03-12 16:59:17 -07:00

sparse_vector_chain.mlir

[mlir][sparse] support sparsifying sparse kernels to sparse-iterator-based loop (#95858)

2024-06-17 16:50:12 -07:00

sparse_vector_concat.mlir

…

sparse_vector_index.mlir

[mlir][sparse] support sparsifying sparse kernels to sparse-iterator-based loop (#95858)

2024-06-17 16:50:12 -07:00

sparse_vector_mv.mlir

[mlir][sparse] made sparse vectorizer more robust on position of invariants (#80766)

2024-02-05 16:12:47 -08:00

sparse_vector_ops.mlir

…

sparse_vector_peeled.mlir

…

sparse_vector.mlir

…

specifier_to_llvm.mlir

[mlir][llvm] Implement ConstantLike for ZeroOp, UndefOp, PoisonOp (#93690)

2024-05-30 08:21:08 +02:00

spy_sddmm_bsr.mlir

[mlir][sparse] support sparsifying sparse kernels to sparse-iterator-based loop (#95858)

2024-06-17 16:50:12 -07:00

spy_sddmm.mlir

…

torch_linalg.mlir

[mlir][sparse] ensure [dis]assembler wrapper methods properly inline (#81907)

2024-02-15 11:39:32 -08:00

transform-ops.mlir

…

unsparsifiable_dense_op.mlir

…

unused-tensor.mlir

…

vectorize_reduction.mlir

[mlir][sparse] support sparsifying sparse kernels to sparse-iterator-based loop (#95858)

2024-06-17 16:50:12 -07:00