[Mlir-commits] [mlir] [mlir][sparse] support sparsifying 2:4 block sparsity (PR #71749)

Yin Tong llvmlistbot at llvm.org
Wed Dec 13 19:47:04 PST 2023


Ag-Cu wrote:

Thanks, This looks like an awsome job! I think I should to take a closer look.

By the way, I read a paper named [VENOM: A Vectorized N:M Format for Unleashing the Power of Sparse Tensor Cores](https://arxiv.org/abs/2310.02065), it is about an open source sparse library for any N:M Format instead of 2:4, and it can still make use of Sparse Tensor Cores. Do you think it is necessary to support N:M Format instead of only NV_24?(Due to current hardware limitations, this matter does not seem to be urgent? I think...)

We are trying to optimize this library, and later we are interested in MLIR's codegen to call Sparse Tensor Cores. If N:M Format is necessary, we may consider adding it to the work plan. And is there any information here that can help develop codegen?

Thanks for your time.

https://github.com/llvm/llvm-project/pull/71749


More information about the Mlir-commits mailing list