PR #28782: [XLA:GPU] Annotate cuBLAS/cuDNN outputs to avoid initcheck failures #3361
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 #28782: [XLA:GPU] Annotate cuBLAS/cuDNN outputs to avoid initcheck failures
Imported from GitHub PR openxla/xla#28782
Upgrades NVTX to v3.2.1 and marks the outputs of cuBLAS/cuDNN as initialized (as compute-sanitizer may emit false positives for kernels using TMA).
Copybara import of the project:
--
55977057d4c3bc3008649cdedc7ddb7923780958 by Sergey Kozub [email protected]:
[XLA:GPU] Annotate cuBLAS/cuDNN outputs to avoid initcheck failures
Merging this change closes #28782
FUTURE_COPYBARA_INTEGRATE_REVIEW=openxla/xla#28782 from openxla:skozub/nvtx_init_annotation 55977057d4c3bc3008649cdedc7ddb7923780958