llama.cpp
vulkan: add tensor-parallel allreduce function
#25051
Merged

vulkan: add tensor-parallel allreduce function #25051

ggerganov merged 21 commits into ggml-org:master from pwilkin:vulkan-tp-p2p
pwilkin
pwilkin pwilkin requested a review 131 days ago
pwilkin pwilkin requested a review from JohannesGaessler JohannesGaessler 131 days ago
pwilkin pwilkin requested a review from jeffbolznv jeffbolznv 131 days ago
pwilkin pwilkin requested a review from 0cc4m 0cc4m 131 days ago
github-actions github-actions added Vulkan
github-actions github-actions added ggml
pwilkin pwilkin marked this pull request as draft 131 days ago
pwilkin pwilkin force pushed to 68c289bd 131 days ago
pwilkin
characharm
wizardeur
netrunnereve
netrunnereve commented on 2026-06-27
digitalscream
cattivik66
marksverdhei
maxious
pwilkin
0cc4m 0cc4m changed the title vulkan: make TP viable vulkan: add allreduce function and fix Tensor Parallel crash 131 days ago
pwilkin pwilkin changed the title vulkan: add allreduce function and fix Tensor Parallel crash vulkan: add allreduce function with cross-device CPU proxy and fix Tensor Parallel crash 131 days ago
pwilkin pwilkin changed the title vulkan: add allreduce function with cross-device CPU proxy and fix Tensor Parallel crash vulkan: add allreduce function with cross-device CPU proxy and fix Tensor Parallel crash [EXPERIMENTAL] 131 days ago
pwilkin
AbdullahMPrograms
digitalscream
pwilkin
AbdullahMPrograms
AbdullahMPrograms
digitalscream
AbdullahMPrograms
wizardeur
pwilkin
digitalscream
pwilkin
pwilkin
digitalscream
pwilkin
AbdullahMPrograms
digitalscream
pwilkin
digitalscream
AbdullahMPrograms
TheBlueMatt
TheBlueMatt commented on 2026-06-28
pwilkin
digitalscream
netrunnereve
netrunnereve
netrunnereve commented on 2026-06-28
TheBlueMatt
pwilkin
TheBlueMatt
0cc4m
pwilkin
En3Tho
0cc4m
pwilkin pwilkin force pushed to 13ac9015 129 days ago
JohannesGaessler
JohannesGaessler commented on 2026-06-29
pwilkin
JohannesGaessler
pwilkin
TheBlueMatt
TheBlueMatt
JohannesGaessler
pwilkin
AbdullahMPrograms
leapdragon
pwilkin
ccarokh
0cc4m
pwilkin pwilkin force pushed from 13ac9015 to 29395e39 43 days ago
0cc4m
0cc4m commented on 2026-09-23
0cc4m
pwilkin
pwilkin pwilkin marked this pull request as ready for review 30 days ago
pwilkin
pwilkin
0cc4m 0cc4m changed the title vulkan: add allreduce function with cross-device CPU proxy and fix Tensor Parallel crash [EXPERIMENTAL] vulkan: add tensor-parallel allreduce function 28 days ago
pwilkin vulkan: add tensor-parallel all-reduce communication
a23aac8f
pwilkin vulkan: require shared semaphores for tensor communication
c42f2a3d
pwilkin vulkan: reduce directly from shared host buffers
a52c1ee6
pwilkin vulkan: fix ring all-reduce races with 3+ devices
b700ea79
pwilkin vulkan: fold the tensor-parallel all-reduce into the compute submissions
bed9ef92
0cc4m - Two-phase all-reduce over shared host memory: write own slot + sign…
3f71382d
0cc4m add minimal submit split size
853cf28d
0cc4m replace opaque_fd with sync_fd for spec compliance
d87d9a3c
0cc4m add precision check
f3681626
0cc4m add semaphore pool
5a4b2934
0cc4m rename env var, add logging if requirements not met
883fe50b
0cc4m reduce comments
c28174c3
0cc4m code improvements
f7563967
pwilkin vulkan: use posix_memalign for the tensor-parallel host buffers
468d073b
pwilkin pwilkin force pushed from c93decc7 to 468d073b 12 days ago
0cc4m
pwilkin vulkan: synchronize the tensor-parallel all-reduce through the host
37f83e6d
netrunnereve
0cc4m
0cc4m use sync_fd on mesa drivers
7d8c3496
0cc4m refactor for readability
128a849c
0cc4m move sync_fd semaphore handling to thread, trigger mesa submit threads
d8446f0f
0cc4m move allreduce functions to new cpp file
b5145b27
0cc4m clean comments
ecfee2ec
0cc4m
Zgate735
0cc4m
jeffbolznv
0cc4m smaller fixes, code improvements
e5793a11
0cc4m
0cc4m approved these changes on 2026-10-30
0cc4m 0cc4m added merge ready
ggerganov ggerganov merged dcbb0916 into master 3 days ago
ngxson
sasa7812
0cc4m
sasa7812
sasa7812

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone