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

OperationProfiler and PerseusOptimizer server and client #21

Open
jaywonchung opened this issue Oct 8, 2023 · 1 comment
Open

OperationProfiler and PerseusOptimizer server and client #21

jaywonchung opened this issue Oct 8, 2023 · 1 comment
Labels
enhancement New feature or request

Comments

@jaywonchung
Copy link
Member

Perseus is an energy scheduler for large model training (although we're looking into applying this for large model inference, too).

Perseus requires the time and energy consumption profiling results of each forward and backward computations in each pipeline stage in order to schedule energy with lowtime. That's what OperationProfiler will do.

The PerseusOptimizer server will, for now, receive a Python file that lists GPU frequencies (produced by lowtime) and instruct the PerseusOptimizer client (integrated into the user's training framework) to change GPU frequencies. The server-client split is beneficial in order for Perseus to be agnostic to the training framework. Otherwise, energy scheduling (which requires a holistic view of all computations that happen across all ranks, i.e. the "policy") and the method of realizing the energy schedule in a distributed fashion (i.e., the "mechanism") end up being coupled.

@jaywonchung jaywonchung added the enhancement New feature or request label Oct 8, 2023
@jaywonchung
Copy link
Member Author

PerseusOptimizer was implemented as PipelineFrequencyOptimizer but OperationProfiler hasn't been implemented yet.

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