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

Decouple ray submitter, worker, and head resources #5933

Merged
merged 1 commit into from
Nov 12, 2024

Conversation

Sovietaced
Copy link
Contributor

@Sovietaced Sovietaced commented Oct 29, 2024

Tracking issue

Related to #5666

Why are the changes needed?

If you try to use the Ray plugin for GPU related workloads the submitter pod ends up consuming as many GPUs as the heads/worker nodes.

What changes were proposed in this pull request?

This pull request introduces resources fields in the worker and head node specs. It then uses the task resources for the submitter pod.

How was this patch tested?

Unit tests

Setup process

Screenshots

Check all the applicable boxes

  • I updated the documentation accordingly.
  • All new and existing tests passed.
  • All commits are signed-off.

Related PRs

flytekit PR will come

Docs link

TBA

@Sovietaced Sovietaced force-pushed the issue-5666 branch 2 times, most recently from 146c464 to 75c0ceb Compare October 29, 2024 03:15
@Sovietaced Sovietaced marked this pull request as ready for review October 29, 2024 03:16
EngHabu
EngHabu previously approved these changes Oct 29, 2024
Copy link
Contributor

@EngHabu EngHabu left a comment

Choose a reason for hiding this comment

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

Thank you for this!

Copy link

codecov bot commented Oct 29, 2024

Codecov Report

Attention: Patch coverage is 58.10811% with 31 lines in your changes missing coverage. Please review.

Project coverage is 36.95%. Comparing base (3409ae1) to head (2ab33a1).
Report is 1 commits behind head on master.

Files with missing lines Patch % Lines
flyteplugins/go/tasks/plugins/k8s/ray/ray.go 67.18% 14 Missing and 7 partials ⚠️
flyteidl/gen/pb-go/flyteidl/plugins/ray.pb.go 0.00% 10 Missing ⚠️
Additional details and impacted files
@@            Coverage Diff             @@
##           master    #5933      +/-   ##
==========================================
- Coverage   36.95%   36.95%   -0.01%     
==========================================
  Files        1310     1310              
  Lines      131401   131451      +50     
==========================================
+ Hits        48556    48574      +18     
- Misses      78631    78656      +25     
- Partials     4214     4221       +7     
Flag Coverage Δ
unittests-datacatalog 51.58% <ø> (ø)
unittests-flyteadmin 54.07% <ø> (ø)
unittests-flytecopilot 22.23% <ø> (ø)
unittests-flytectl 62.39% <ø> (ø)
unittests-flyteidl 6.95% <0.00%> (-0.01%) ⬇️
unittests-flyteplugins 53.83% <67.18%> (-0.02%) ⬇️
unittests-flytepropeller 43.08% <ø> (ø)
unittests-flytestdlib 55.30% <ø> (-0.02%) ⬇️

Flags with carried forward coverage won't be shown. Click here to find out more.

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

Copy link
Contributor

@eapolinario eapolinario left a comment

Choose a reason for hiding this comment

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

Awesome. Thank you!

@Sovietaced Sovietaced merged commit 3c3ae05 into flyteorg:master Nov 12, 2024
51 checks passed
@Sovietaced Sovietaced deleted the issue-5666 branch November 12, 2024 17:03
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.

4 participants