Skip to content

perf(kda): avoid the Prefill convolution layout copy - #1704

Draft
shihaobai wants to merge 1 commit into
mainfrom
bsh/glm-prefill-kda-token-major-conv
Draft

shihaobai wants to merge 1 commit into
mainfrom
bsh/glm-prefill-kda-token-major-conv

Conversation

@shihaobai

Copy link
Copy Markdown
Collaborator

Read GLM Prefill QKV directly in token-major order on supported Hopper batches. Commit the convolution history after all tiles have read it. Enable with LIGHTLLM_GLM53_PREFILL_TOKEN_MAJOR_CONV=1; other paths keep the existing convolution.

The delivery build preserved output and state bitwise in seven H100 cases. The main port passes Black, Flake8, and ten host dispatch cases. Main CUDA regression tests are included; the rerun is pending.

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