New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Set cuda device max connections based on cuda capability #451

Closed

guyueh1 wants to merge 2 commits into NVIDIA:main from guyueh1:set_cuda_device_max_conn_from_sm_arch

Contributor

guyueh1 commented Dec 20, 2024

Set cuda device max connections based on cuda capability.

do not set when there is no TP and CP
when there is TP or CP

set to 32 if cuda capability >=10
set to 1 if cuda capability < 10

Guyue Huang added 2 commits

December 20, 2024 09:55


          Set cuda device max connections based on cuda capability

5ea86a0

Signed-off-by: Guyue Huang <[email protected]>


          Add get_cuda_device_max_connections.yaml

cf67cc7

Signed-off-by: Guyue Huang <[email protected]>

guyueh1 closed this

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet