Skip to content

[CUDA] Support 3/5/6-bit quants in QMV#3236

Merged
zcbenz merged 1 commit intoml-explore:mainfrom
zcbenz:qmv-356-bits
Mar 10, 2026
Merged

[CUDA] Support 3/5/6-bit quants in QMV#3236
zcbenz merged 1 commit intoml-explore:mainfrom
zcbenz:qmv-356-bits

Conversation

@zcbenz
Copy link
Collaborator

@zcbenz zcbenz commented Mar 10, 2026

Refs #2536.

This PR adds some cutlass::NumericArrayConverter specializations for converting 3/5/6-bit ints to float, which I will move into a header (like cutlass_utils.cuh) when adding QMM implementations.

Copy link
Member

@angeloskath angeloskath left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Looks great, very clean!

@zcbenz zcbenz merged commit 9d03a1b into ml-explore:main Mar 10, 2026
16 checks passed
@zcbenz zcbenz deleted the qmv-356-bits branch March 10, 2026 10:09
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants