Skip to content

Conversation

@tianleiwu
Copy link
Contributor

Enhance MatMulNBits CUDA kernel testing:
(1) Add a kernel testing for different cuda kernels used in MatMulNBits.
(2) Refactoring the gemm profiler to use cuda allocator
(2) Add verbose logging macros.
(3) Adjustments to speed up compiling when sm90 is excluded from build.

Example kernel test output:
image

@tianleiwu tianleiwu marked this pull request as draft June 18, 2025 22:03
@tianleiwu tianleiwu marked this pull request as ready for review June 19, 2025 20:38
@tianleiwu tianleiwu merged commit 7268117 into main Jun 20, 2025
89 checks passed
@tianleiwu tianleiwu deleted the tlwu/fpA_intB_gemm_kernel_test branch June 20, 2025 20:45
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.

4 participants