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

Refactor find_matches_for_node return values #920

Merged
merged 1 commit into from
Apr 10, 2024

Conversation

cindyyuanjiang
Copy link
Collaborator

@cindyyuanjiang cindyyuanjiang commented Apr 10, 2024

find_matches_for_node returns a mapping of CPU to GPU machines and a mapping of supported GPUs. It has 5 use sites, but 4 of them only need one of the mappings.

This PR refactors the function return values. For onprem, the implementation can directly get supported_mc_map without running find_matches_for_node.

Signed-off-by: cindyyuanjiang <[email protected]>
@cindyyuanjiang cindyyuanjiang self-assigned this Apr 10, 2024
@cindyyuanjiang cindyyuanjiang requested review from amahussein, nartal1 and parthosa and removed request for amahussein April 10, 2024 00:14
@cindyyuanjiang cindyyuanjiang added the user_tools Scope the wrapper module running CSP, QualX, and reports (python) label Apr 10, 2024
Copy link
Collaborator

@nartal1 nartal1 left a comment

Choose a reason for hiding this comment

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

Thanks @cindyyuanjiang for this refactor. LGTM.

Copy link
Collaborator

@parthosa parthosa left a comment

Choose a reason for hiding this comment

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

Thanks @cindyyuanjiang. LGTM.

@cindyyuanjiang cindyyuanjiang merged commit 5df1c3f into NVIDIA:dev Apr 10, 2024
16 checks passed
@cindyyuanjiang cindyyuanjiang deleted the refactor-function branch April 10, 2024 17:45
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
user_tools Scope the wrapper module running CSP, QualX, and reports (python)
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants