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

feat: Add VRAM and disk space matching #471

Merged
merged 7 commits into from
Dec 18, 2024

Conversation

bgins
Copy link
Contributor

@bgins bgins commented Dec 13, 2024

Summary

This pull request makes the following changes:

  • Add VRAM to matching algorithm
  • Add disk space to matching algorithm

We want the Solver matching algorithm to match on minimum VRAM and disk space. This pull request implements these constraints.

Task/Issue reference

Closes: #470

Test plan

We have added new test cases for VRAM and disk space to our matcher test suite. Run the tests with:

go test ./pkg/solver/matcher --tags="unit" --count 1 -v

Start the stack. Run a job. Our usual test procedure ensures this pull request does not break anything, but note that it does not exercise cases where a job creator requests VRAM or disk space.

Details

Resource providers currently report VRAM and disk space in resource offers.

The job offer includes a machine spec, but we have not implemented CLI options for setting GPUs (with VRAM) or disk space. As a result, GPUs and disk space use default zero values, empty slice and 0 respectively.

Related issues or PRs

Epic: https://github.com/Lilypad-Tech/internal/issues/367

@bgins bgins requested a review from a team as a code owner December 13, 2024 18:01
@cla-bot cla-bot bot added the cla-signed label Dec 13, 2024
@bgins bgins changed the title Bgins/feat add vram and disk matching feat: Add vram and disk matching Dec 13, 2024
@bgins bgins changed the title feat: Add vram and disk matching feat: Add VRAM and disk space matching Dec 13, 2024
@bgins bgins force-pushed the bgins/feat-add-vram-and-disk-matching branch from d0384bd to 33a3bc0 Compare December 13, 2024 19:15
@bgins bgins self-assigned this Dec 13, 2024
Copy link
Collaborator

@walkah walkah left a comment

Choose a reason for hiding this comment

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

Tested locally. Nicely done! :shipit: 🚀

@bgins bgins merged commit 552d8e1 into main Dec 18, 2024
5 checks passed
@bgins bgins deleted the bgins/feat-add-vram-and-disk-matching branch December 18, 2024 15:57
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Add VRAM and Disk Space to Solver matching algorithm
2 participants