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

Support mixed MIG+MPS dynamic partititioning #28

Open
Telemaco019 opened this issue Mar 31, 2023 · 0 comments
Open

Support mixed MIG+MPS dynamic partititioning #28

Telemaco019 opened this issue Mar 31, 2023 · 0 comments
Labels
enhancement New feature or request

Comments

@Telemaco019
Copy link
Member

Description

Currently, when enabling Dynamic GPU Partitioning on a node, it is possible to choose only between MIG or MPS by adding one of the following labels: nos.nebuly.com/gpu-partitioning: "mig" or nos.nebuly.com/gpu-partitioning: "mps".

It would be nice to have a third dynamic partitioning option that mixes MIG and MPS. This would be particularly useful for further partitioning MIG devices with MPS, as often the smallest available MIG device on a GPU is way larger than the resources required by the workloads.

For instance, the smallest MIG profile for NVIDIA-A100-SXM4-80GB is 1g.10gb, which provides 10GB of GPU memory. However, since many workloads require less than 10GB of GPU memory, this leads to inefficiencies.

Right now the alternative is to partition GPUs using MPS, which allows the creation of GPU slices of arbitrary size. However, MPS does not provide full workload isolation. Using MPS on top of MIG would enable finer-grained partitioning without compromising too much workload isolation, as only the workloads sharing the same MIG partition wouldn't be fully isolated.

Proposed solution

Add the possibility to label a node with nos.nebuly.com/gpu-partitioning: "mixed". For nodes with this label, nos should automatically use MPS for partitioning the smallest available MIG devices according to the requested resources.

@Telemaco019 Telemaco019 added the enhancement New feature or request label Mar 31, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

No branches or pull requests

1 participant