Skip to content

fix: use vllm_nccl installed nccl version#26

Merged
njhill merged 1 commit intomainfrom
ubi-vllm-nccl
May 13, 2024
Merged

fix: use vllm_nccl installed nccl version#26
njhill merged 1 commit intomainfrom
ubi-vllm-nccl

Conversation

@tjohnson31415
Copy link
Member

Install and configure use of the NCCL version recommended by vLLM via the vllm-nccl package. The install is a little wonky... but this set of changes should work.

Signed-off-by: Travis Johnson <tsjohnso@us.ibm.com>
Copy link
Contributor

@njhill njhill left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks @tjohnson31415!

@njhill njhill merged commit 21fb852 into main May 13, 2024
@njhill njhill deleted the ubi-vllm-nccl branch May 13, 2024 21:23
tdoublep pushed a commit that referenced this pull request Jan 20, 2025
…fig settings (#26)

This PR refers to the following updates.
- Replaced aiu_common.py with aiu_setup.py from aiu-app-sw-tracker repo
- Moved aiu_setup.py to init_device() in sendnn_worker.py
- Updated various torch._dynamo.config settings in sendnn.py

---------

Signed-off-by: Nikolaos Papandreou <npo@zurich.ibm.com>
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.

2 participants