Skip to content

Commit 16c3bf4

Browse files
authored
ci: install NCCL on Linux CUDA runners (#538)
Enables multi-GPU allreduce; falls back to internal pipeline if unavailable.
1 parent 56629e9 commit 16c3bf4

1 file changed

Lines changed: 7 additions & 0 deletions

File tree

.github/workflows/menlo-build.yml

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -406,6 +406,13 @@ jobs:
406406
ccache -V
407407
rm -rf /tmp/ccache-4.10.2-linux-x86_64.tar.xz /tmp/ccache-4.10.2-linux-x86_64
408408
409+
- name: Install NCCL (Linux CUDA)
410+
if: runner.os == 'Linux' && contains(matrix.runs-on, 'cuda')
411+
continue-on-error: true
412+
run: |
413+
sudo apt-get install -y libnccl2 libnccl-dev || \
414+
echo "::warning::libnccl-dev not available via apt; multi-GPU allreduce will fall back to the internal pipeline"
415+
409416
- name: Set up homebrew path on macOS
410417
if: runner.os == 'macOS'
411418
run: |

0 commit comments

Comments
 (0)