vllm
[Kernel] Porting the TRTLLM minimax_allreduce_rms kernels
#37045
Merged
Overview
Commits
50
Changes
View On GitHub

[Kernel] Porting the TRTLLM minimax_allreduce_rms kernels #37045

youkaichao merged 50 commits into vllm-project:main from jeejeelee:minmax-m2-norm-trtllm-kernel
jeejeelee
jeejeelee Init
24407645
jeejeelee Backup
4e1712ba
jeejeelee Generation LGTM
2788b448
jeejeelee Backup
10323a90
jeejeelee Support torch.compile
faac03c8
jeejeelee jeejeelee requested a review from tlrmchlsmth tlrmchlsmth 140 days ago
jeejeelee jeejeelee requested a review from LucasWilkinson LucasWilkinson 140 days ago
jeejeelee jeejeelee requested a review from tdoublep tdoublep 140 days ago
jeejeelee jeejeelee marked this pull request as draft 140 days ago
mergify mergify added ci/build
gemini-code-assist
gemini-code-assist commented on 2026-03-14
jeejeelee cleanup code
1683c3d2
jeejeelee Add testing
7ba290a6
jeejeelee Fix
69f231c7
jeejeelee Merge branch 'main' into minmax-m2-norm-trtllm-kernel
6ac265df
jeejeelee Cleanup
d1dce0bf
jeejeelee Merge branch 'main' into minmax-m2-norm-trtllm-kernel
8112fbcf
jeejeelee Modify logic
db534879
jeejeelee Skip qk copy
83f6d1a5
jeejeelee Modify kernel input
dc9d4276
jeejeelee Merge branch 'main' into minmax-m2-norm-trtllm-kernel
9066f23a
jeejeelee Move
8cec1815
jeejeelee jeejeelee marked this pull request as ready for review 135 days ago
jeejeelee jeejeelee requested a review from mgoin mgoin 135 days ago
jeejeelee jeejeelee requested a review from WoosukKwon WoosukKwon 135 days ago
jeejeelee jeejeelee requested a review from yewentao256 yewentao256 135 days ago
mergify
yewentao256
yewentao256 commented on 2026-03-19
yewentao256 yewentao256 added ready
jeejeelee Cleanup
917de772
jeejeelee Move
f4355378
jeejeelee jeejeelee force pushed to f4355378 135 days ago
jeejeelee Merge branch 'main' into minmax-m2-norm-trtllm-kernel
ae3661e4
jeejeelee Move
e30277b6
jeejeelee jeejeelee removed ready
jeejeelee Merge branch 'main' into minmax-m2-norm-trtllm-kernel
83ec0393
jeejeelee Merge branch 'main' into minmax-m2-norm-trtllm-kernel
9c53b60d
jeejeelee FIX
2556d521
jeejeelee Merge branch 'main' into minmax-m2-norm-trtllm-kernel
2614b473
jeejeelee jeejeelee added ready
jeejeelee Merge branch 'main' into minmax-m2-norm-trtllm-kernel
a547d685
jeejeelee jeejeelee requested a review from yewentao256 yewentao256 128 days ago
tjtanaa
tjtanaa commented on 2026-03-26
tjtanaa tjtanaa added rocm
tjtanaa
tjtanaa commented on 2026-03-26
jeejeelee Merge branch 'main' into minmax-m2-norm-trtllm-kernel
8b894887
jeejeelee Merge branch 'main' into minmax-m2-norm-trtllm-kernel
dea46412
jeejeelee Merge branch 'main' into minmax-m2-norm-trtllm-kernel
62a5f66d
jeejeelee Copy latest imp
cd4d78ff
jeejeelee jeejeelee requested a review from tomeras91 tomeras91 124 days ago
mergify
wzhao18
jeejeelee Move
16d65441
jeejeelee Add custom op
a3f01e76
jeejeelee move
7e6aeb7e
jeejeelee move
1822e545
jeejeelee Kernel fusion
68b74c05
jeejeelee Kernel fusion done
cff5dcbb
jeejeelee Merge remote-tracking branch 'origin/main' into minmax-m2-norm-trtllm…
9efeba5f
jeejeelee Fix
b98a1e16
jeejeelee Backup
fee9e114
jeejeelee Delete redundant code
db28d8df
jeejeelee jeejeelee requested a review from youkaichao youkaichao 117 days ago
jeejeelee jeejeelee requested a review from robertgshaw2-redhat robertgshaw2-redhat 117 days ago
jeejeelee jeejeelee requested a review from houseroad houseroad 117 days ago
jeejeelee jeejeelee requested a review from hmellor hmellor 117 days ago
jeejeelee jeejeelee requested a review from ProExpertProg ProExpertProg 117 days ago
jeejeelee jeejeelee requested a review from zou3519 zou3519 117 days ago
jeejeelee jeejeelee requested a review from BoyuanFeng BoyuanFeng 117 days ago
jeejeelee jeejeelee requested a review from vadiklyutiy vadiklyutiy 117 days ago
jeejeelee jeejeelee removed ready
jeejeelee Fix test
4bd3eada
ywang96 ywang96 added ready
ywang96
jeejeelee FIX
2e63a801
jeejeelee jeejeelee force pushed to 2e63a801 116 days ago
khluu khluu added this to the v0.19.1 cherry picks milestone 116 days ago
wzhao18
wzhao18 commented on 2026-04-07
wzhao18
wzhao18 commented on 2026-04-07
wzhao18
jeejeelee Merge branch 'main' into minmax-m2-norm-trtllm-kernel
c9328b23
jeejeelee Revert model
efa99565
jeejeelee Fix
1eaca8a0
mergify
jeejeelee Fix
741a7363
mergify
mergify mergify added needs-rebase
jeejeelee Fix
685f22cf
jeejeelee Merge branch 'main' into minmax-m2-norm-trtllm-kernel
9c225e13
mergify mergify removed needs-rebase
jeejeelee Merge branch 'main' into minmax-m2-norm-trtllm-kernel
c9e8c3f6
jeejeelee Fix
0b917927
jeejeelee FIX
8c9e80e0
youkaichao
youkaichao approved these changes on 2026-04-10
youkaichao youkaichao merged ecd1ea13 into main 113 days ago
jeejeelee jeejeelee deleted the minmax-m2-norm-trtllm-kernel branch 112 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
rocm ready ci/build
Milestone
v0.19.1 cherry picks
Terms Of Use Privacy Policy Refund Policy Impressum