PR #13603: NVTX: name threads, CUDA devices and CUDA streams #2337
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
![copybara-service[bot]](https://github.com/avatars/in/44061?s=60&v=4){kind=small}
Imported from GitHub PR openxla/xla#13603 This aims to improve the profiling experience. These names are shown in the Nsight Systems UI. Device names:  Stream names:  Thread names:  This also provides a missing link between replica IDs in the HLO and the physical devices in the profile. Copybara import of the project: -- 12a02b67bd9db8b3f69ba1e0d00c7881f767f037 by Olli Lupton <olupton@nvidia.com>: NVTX: name threads, CUDA devices and CUDA streams -- bdf8dbf7700cbe7ce72070c25ce3d21e2dfeb54f by Olli Lupton <olupton@nvidia.com>: Add missing header -- 98a80a40add79f108cb89987724c35f82cd727e4 by Olli Lupton <olupton@nvidia.com>: add stubs Merging this change closes #13603 FUTURE_COPYBARA_INTEGRATE_REVIEW=openxla/xla#13603 from olupton:name-devices-streams-and-threads 98a80a40add79f108cb89987724c35f82cd727e4 PiperOrigin-RevId: 643290582
copybara-service
bot
force-pushed
the
test_643290582
branch
from
b429874
to
4d865bf
Compare
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
PR #13603: NVTX: name threads, CUDA devices and CUDA streams
Imported from GitHub PR openxla/xla#13603
This aims to improve the profiling experience. These names are shown in the Nsight Systems UI.
Device names:
Stream names:
Thread names:
This also provides a missing link between replica IDs in the HLO and the physical devices in the profile.
Copybara import of the project:
--
12a02b67bd9db8b3f69ba1e0d00c7881f767f037 by Olli Lupton olupton@nvidia.com:
NVTX: name threads, CUDA devices and CUDA streams
--
bdf8dbf7700cbe7ce72070c25ce3d21e2dfeb54f by Olli Lupton olupton@nvidia.com:
Add missing header
--
98a80a40add79f108cb89987724c35f82cd727e4 by Olli Lupton olupton@nvidia.com:
add stubs
Merging this change closes #13603
FUTURE_COPYBARA_INTEGRATE_REVIEW=openxla/xla#13603 from olupton:name-devices-streams-and-threads 98a80a40add79f108cb89987724c35f82cd727e4