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.

Sign up for GitHub

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

Jump to bottom

Integrate Pytorch Profiler into Modulus, including StaticCapture* #750

Open

coreyjadams opened this issue Jan 7, 2025 · 0 comments

Assignees

Collaborator

coreyjadams commented Jan 7, 2025

Enable Profiler in two modes in modulus:

Standalone, provide default configurations and both context and decorator functionality. This will let users quickly slap a profiler on their code, with useful configuration enabled.
Integrate into StaticCaptureTrain and StaticCaptureEval: users employing these tools should be able to capture profiles without leaving that interface.

Modulus will also provide some basic configurations for the profiling (lightweight, CPU-only, full, nvtx-annotations, etc).

Profiler must work in both distributed and non-distributed mode.

coreyjadams self-assigned this

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment