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

DynamicRayRunner #412

Merged
merged 18 commits into from
Jan 4, 2023
Merged

DynamicRayRunner #412

merged 18 commits into from
Jan 4, 2023

Conversation

xcharleslin
Copy link
Contributor

@xcharleslin xcharleslin commented Dec 30, 2022

A working Ray Runner that uses DynamicSchedule.

Slight performance regression against existing RayRunner, likely since this new runner needs partition metadata and makes a second ray.get call per partition to get it.

@xcharleslin xcharleslin marked this pull request as ready for review December 31, 2022 07:59
@xcharleslin xcharleslin enabled auto-merge (squash) January 4, 2023 19:47
@xcharleslin xcharleslin merged commit d05e9c4 into main Jan 4, 2023
@xcharleslin xcharleslin deleted the charles/rayrunner2 branch January 4, 2023 19:53
xcharleslin added a commit that referenced this pull request Jan 5, 2023
…r task management (#418)

Optimizations in #412:

- Zero dynamic ray.remote registrations (they all happen at module
initialization). This is a strict improvement over RayRunner.

Optimizations in this PR:

- Metadata is computed and retrieved as needed, instead of for all
partitions. This brings down metadata overhead vs RayRunner to
negligible.
- Drop asyncio and move entirely to ray.wait for task management. This
should correctly minimize idle time by the dispatcher.
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.

None yet

1 participant