Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Usage]: fp8 sparse gemm in vllm/csrc/sparse/cutlass/sparse_scaled_mm__xxx #12344

Open
1 task done
zhink opened this issue Jan 23, 2025 · 0 comments
Open
1 task done
Labels
usage How to use vllm

Comments

@zhink
Copy link

zhink commented Jan 23, 2025

Your current environment

The output of `python collect_env.py`

How would you like to use vllm

In the three implementation files of fp8 sparse gemm vllm/csrc/sparse/cutlass/sparse_scaled_mm__xxx.

  1. Can you abstract the implementation formula of cutlass_scaled_sparse_mm_sm90?
    1. In addition, I understand that bt_nzs bt_meta is the data after gemm weight sparse (from sparse_compressor_xxx), what are the arrangements of the three tensors bt_nzs, a and c respectively (in ColumnMajor RowMajor)?

Before submitting a new issue...

  • Make sure you already searched for relevant issues, and asked the chatbot living at the bottom right corner of the documentation page, which can answer lots of frequently asked questions.
@zhink zhink added the usage How to use vllm label Jan 23, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
usage How to use vllm
Projects
None yet
Development

No branches or pull requests

1 participant