Skip to content

nsys global-step profiling support#702

Open
chenrui17 wants to merge 2 commits into
nvidia-cosmos:mainfrom
chenrui17:nsys-support
Open

nsys global-step profiling support#702
chenrui17 wants to merge 2 commits into
nvidia-cosmos:mainfrom
chenrui17:nsys-support

Conversation

@chenrui17

Copy link
Copy Markdown

Add Nsight Systems profiling support for policy and rollout workers

Introduce NVTX annotations and controller-orchestrated Nsight capture windows so policy and rollout profiles can be collected over the same global training steps without changing training computation logic.

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.

1 participant