Skip to content

Conversation

@olupton
Copy link
Collaborator

@olupton olupton commented Mar 10, 2025

  • Repeated entries in XLA_FLAGS will no longer cause assertion errors. Only the last flag will be passed to the child application.
  • nsys-jax-patch-nsys now uses [ugly] logic that works even for cuda-nsight-systems packages where nsys in $PATH is a shim
    • CI tests are added for this case

olupton added 5 commits March 10, 2025 12:33
The cuda-nsight-systems-MAJOR-MINOR packages produce an installation
where nsys is a shim script, rather than the actual executable.
The last value wins. Earlier values are pruned from the XLA_FLAGS passed
to the profiled application.
@olupton olupton requested a review from Steboss March 11, 2025 07:24
Copy link
Contributor

@Steboss Steboss left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@olupton olupton merged commit ee46a55 into main Mar 13, 2025
119 of 123 checks passed
@olupton olupton deleted the olupton/nsys-jax-improvements branch March 13, 2025 10:35
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.

3 participants