Skip to content
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.

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

[FluidStack] Add Nvlink Accelerator Mappings #3467

Conversation

mjibril
Copy link
Contributor

@mjibril mjibril commented Apr 23, 2024

  • Add A100-80GB and H100 Nvlink accelerator mappings

  • Remove custom/configurable plans

Changes Description

  • New gpu types mapped to Skypilot types
  • Removed unsupported custom plans

Tests ran

sky launch
python sky/clouds/service_catalog/data_fetchers/fetch_fluidstack.py

Tested (run the relevant ones):

  • [* ] Code formatting: bash format.sh
  • [* ] Any manual or new tests for this PR (please specify below)
    • sky launch
  • All smoke tests: pytest tests/test_smoke.py
  • Relevant individual smoke tests: pytest tests/test_smoke.py::test_fill_in_the_name
  • Backward compatibility tests: bash tests/backward_comaptibility_tests.sh

Copy link
Collaborator

@Michaelvll Michaelvll left a comment

Choose a reason for hiding this comment

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

Thanks for updating the mapping @mjibril! The changes look good to me.

Comment on lines +21 to +22
'H100_NVLINK_80GB': 'H100',
'A100_NVLINK_80GB': 'A100-80GB',
Copy link
Collaborator

Choose a reason for hiding this comment

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

Nice! It seems I getting the following "failed to map" list, do we want to add them as well?

Could not map QUADRO_RTX_5000_16GB
Could not map RTX_A4500_16GB
Could not map QUADRO_RTX_4000_12GB
Could not map RTX_A6000_24GB
Could not map Quadro_P5000_12GB
Could not map QUADRO_P5000_16GB

Copy link
Contributor Author

Choose a reason for hiding this comment

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

Not really. I will suppress these.

* Add A100-80GB and H100 Nvlink accelerator mappings

* Remove custom/configurable plans
@mjibril mjibril force-pushed the fluidstack-add-accelerator-mappings branch from 39501ff to 3b8133c Compare April 23, 2024 19:43
@Michaelvll Michaelvll merged commit f3be773 into skypilot-org:master Apr 24, 2024
20 checks passed
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