Skip to content

support deepgemm new api#3827

Merged
windreamer merged 1 commit intoInternLM:mainfrom
grimoire:support-deepgemm2
Aug 12, 2025
Merged

support deepgemm new api#3827
windreamer merged 1 commit intoInternLM:mainfrom
grimoire:support-deepgemm2

Conversation

@grimoire
Copy link
Copy Markdown
Collaborator

DeepGemm update it's python api since PR112

This PR add third-party wrap of deep-gemm. Support both 1.0 and 2.0.

Since deepgemm2.0 used cpp wrapped compile tools, logger can not be patched in it. If compile log is required, DG_JIT_PRINT_COMPILER_COMMAND=1 or DG_JIT_DEBUG=1 can be used.

@windreamer windreamer merged commit 0c661c9 into InternLM:main Aug 12, 2025
5 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants