Skip to content

[Debug] support using deemgemm bf16 gmm, do not merge#1343

Draft
HIT-cwh wants to merge 1 commit intoInternLM:mainfrom
HIT-cwh:use_deepgemm_bf16
Draft

[Debug] support using deemgemm bf16 gmm, do not merge#1343
HIT-cwh wants to merge 1 commit intoInternLM:mainfrom
HIT-cwh:use_deepgemm_bf16

Conversation

@HIT-cwh
Copy link
Collaborator

@HIT-cwh HIT-cwh commented Dec 8, 2025

Set BF16_GMM_USING_DEEPGEMM to 1 to enable bf16 grouped gemm kernel from deepgemm:

export BF16_GMM_USING_DEEPGEMM=1 

@CyCle1024 CyCle1024 marked this pull request as draft December 9, 2025 04:16
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.

1 participant