Aart Bik e383eaa647 [mlir][sparse] parameterize MTTKRP kernel ...

Rather than hardcoding all constants, we now use the input tensor to drive the
code setup. Of course, we still need to hardcode dim-2 of A and the final
verification in CHECK is input dependent, but overall this sets a slightly
better example of tensor setup in general.

Reviewed By: bixia

Differential Revision: https://reviews.llvm.org/D117349

2022-01-14 16:20:31 -08:00

..

dense_output.mlir

[MLIR] Replace std ops with arith dialect ops

2021-10-13 03:07:03 +00:00

lit.local.cfg

[mlir][sparse] add a "release" operation to sparse tensor dialect

2021-10-05 09:35:59 -07:00

sparse_cast.mlir

[mlir] Move memref.[tensor_load|buffer_cast|clone] to "bufferization" dialect.

2021-11-25 11:50:39 +01:00

sparse_constant_to_sparse_tensor.mlir

[MLIR] Replace std ops with arith dialect ops

2021-10-13 03:07:03 +00:00

sparse_conversion_dyn.mlir

[mlir][sparse] fix conversion bug when changing pointer/index sizes

2021-10-28 17:24:38 -07:00

sparse_conversion_ptr.mlir

[mlir][sparse] refine the mixed width sparse conversion test

2021-10-29 13:31:04 -07:00

sparse_conversion_sparse2dense.mlir

[mlir] Move memref.[tensor_load|buffer_cast|clone] to "bufferization" dialect.

2021-11-25 11:50:39 +01:00

sparse_conversion.mlir

[mlir][sparse] move conversion test back to original CHECK testing

2021-10-28 09:03:26 -07:00

sparse_filter_conv2d.mlir

[mlir] Move memref.[tensor_load|buffer_cast|clone] to "bufferization" dialect.

2021-11-25 11:50:39 +01:00

sparse_flatten.mlir

[mlir] Move memref.[tensor_load|buffer_cast|clone] to "bufferization" dialect.

2021-11-25 11:50:39 +01:00

sparse_matmul.mlir

[mlir][sparse] finalize sparse output in the presence of reductions

2021-12-07 10:54:29 -08:00

sparse_matrix_ops.mlir

[mlir] Move memref.[tensor_load|buffer_cast|clone] to "bufferization" dialect.

2021-11-25 11:50:39 +01:00

sparse_matvec.mlir

[mlir] Move memref.[tensor_load|buffer_cast|clone] to "bufferization" dialect.

2021-11-25 11:50:39 +01:00

sparse_mttkrp.mlir

[mlir][sparse] parameterize MTTKRP kernel

2022-01-14 16:20:31 -08:00

sparse_out_mult_elt.mlir

[mlir][sparse] added sparse out element wise mult integration test

2021-11-30 16:44:38 -08:00

sparse_out_reduction.mlir

[mlir][sparse] fix typos in integration tests

2021-11-30 15:32:20 -08:00

sparse_out_simple.mlir

[mlir][sparse] fix typos in integration tests

2021-11-30 15:32:20 -08:00

sparse_quantized_matmul.mlir

[mlir] Move memref.[tensor_load|buffer_cast|clone] to "bufferization" dialect.

2021-11-25 11:50:39 +01:00

sparse_reductions.mlir

[mlir] Move memref.[tensor_load|buffer_cast|clone] to "bufferization" dialect.

2021-11-25 11:50:39 +01:00

sparse_sampled_matmul.mlir

[mlir] Move memref.[tensor_load|buffer_cast|clone] to "bufferization" dialect.

2021-11-25 11:50:39 +01:00

sparse_sampled_mm_fusion.mlir

[mlir][sparse] reenable asan for sampled mm integration test

2021-12-09 12:07:56 -08:00

sparse_scale.mlir

[mlir][sparse] run more integration tests with and without SIMD

2021-11-05 12:51:38 -07:00

sparse_spmm.mlir

[mlir] Move memref.[tensor_load|buffer_cast|clone] to "bufferization" dialect.

2021-11-25 11:50:39 +01:00

sparse_storage.mlir

[MLIR] Replace std ops with arith dialect ops

2021-10-13 03:07:03 +00:00

sparse_sum.mlir

[mlir] Move memref.[tensor_load|buffer_cast|clone] to "bufferization" dialect.

2021-11-25 11:50:39 +01:00

sparse_tensor_ops.mlir

[mlir][sparse] refine lexicographic insertion to any tensor

2021-11-17 18:08:42 -08:00

sparse_vector_ops.mlir

[mlir][sparse] generalize sparse tensor output implementation

2021-11-29 16:15:53 -08:00