Skip to content

[AIU Profiler] Change activity types#1182

Closed
mcalman wants to merge 3 commits intopytorch:mainfrom
mcalman:aiu-activities
Closed

[AIU Profiler] Change activity types#1182
mcalman wants to merge 3 commits intopytorch:mainfrom
mcalman:aiu-activities

Conversation

@mcalman
Copy link
Contributor

@mcalman mcalman commented Nov 21, 2025

  1. Changes the activity types from cuda_driver and cuda_runtime to privateuse1_driver and privateuse1_runtime respectively.
  2. Switches from cuda_driver to gpu_memset for memset activities so that they are properly accounted for in the Step Time Breakdown in TensorBoard.

Signed-off-by: Max Calman <maxcalman1@gmail.com>
Signed-off-by: Max Calman <maxcalman1@gmail.com>
@meta-codesync
Copy link

meta-codesync bot commented Dec 1, 2025

@malfet has imported this pull request. If you are a Meta employee, you can view this in D88099683.

@facebook-github-bot
Copy link
Contributor

@mcalman has updated the pull request. You must reimport the pull request before landing.

@facebook-github-bot
Copy link
Contributor

@mcalman has updated the pull request. You must reimport the pull request before landing.

@mcalman
Copy link
Contributor Author

mcalman commented Jan 21, 2026

@malfet @aaronenyeshi can we merge this?

#1171, #1172, and #1229 also include other small fixes to the AIU profiler. If possible, can you review these as well?

@meta-codesync
Copy link

meta-codesync bot commented Jan 26, 2026

@divyanshk merged this pull request in 91a42cb.

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

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants