Skip to content

New functionality & documentation improvement in optim_test_tools#12

Merged
xupinjie merged 1 commit intoNVIDIA:mainfrom
RmSchaffert:optim_test_tools_ext
Mar 23, 2026
Merged

New functionality & documentation improvement in optim_test_tools#12
xupinjie merged 1 commit intoNVIDIA:mainfrom
RmSchaffert:optim_test_tools_ext

Conversation

@RmSchaffert
Copy link
Copy Markdown
Collaborator

@RmSchaffert RmSchaffert commented Mar 23, 2026

  • Added numba_nvtx module: Use of NVTX ranges in Numba-compiled code
  • TensorDumper
    • Added PICKLE dump type
    • Added ranges (for disambiguation of tensor names in certain scenarios)
  • Improved documentation

Type of Change

Please select (at least one):

  • Bug fix (non-breaking change that fixes an issue)
  • New feature (non-breaking change that adds functionality)
  • Breaking change (fix or feature that would cause existing functionality to change)
  • Documentation / examples / tutorials / demos
  • Supporting functionality change (fix or feature in documentation generation, helper scripts, ...)
  • Refactoring / internal change
  • Other (please describe):

Testing

Checklist for testing:

  • Tests added or updated if/as needed
  • Repository test runner executed: scripts/run_tests.sh

Optionally, add a brief description.

Documentation, Examples, Tutorials, Demos

Checklist for documentation:

  • User-facing documentation updated if/as needed (including API docs)
  • Examples / tutorials / demos updated or added (if relevant)
  • Limitations and constraints documented (if relevant)
  • Performance documented (if relevant)
  • Documentation building successful & checks outlined in the Documentation Checks section of the Contribution Guide are performed

Optionally, add a brief description.

Code Quality

Checklist for dependencies:

  • Dependencies updated in the relevant pyproject.toml if/as needed
  • Code formatted according to the Code Formatting Guide

Optionally, add a brief description.

Related Issues / Context

If applicable, link related issues, discussions etc.


DCO / Sign-Off

Please refer to the section on Signing Your Work & Developer Certificate of Origin (DCO)
in the Contribution Guide before submitting your contribution.

References

For additional details, please refer to the Contribution Guide.
The following guides are available (referenced in the Contribution Guide for further details):

Please also refer to the summary checklist in the Contribution Guide,
which is a guideline for what to consider when submitting your contribution and covers the same topics as the checklists above.

@RmSchaffert RmSchaffert requested a review from xupinjie March 23, 2026 05:57
- Added `numba_nvtx` module: Use of NVTX ranges in Numba-compiled code
- `TensorDumper`
  - Added `PICKLE` dump type
  - Added ranges (for disambiguation of tensor names in certain scenarios)
- Improved documentation

Signed-off-by: Roman Schaffert <rschaffert@nvidia.com>
@RmSchaffert RmSchaffert force-pushed the optim_test_tools_ext branch from 9018ca7 to 92c6535 Compare March 23, 2026 05:59
@RmSchaffert
Copy link
Copy Markdown
Collaborator Author

/build

@xupinjie xupinjie self-requested a review March 23, 2026 07:28
@xupinjie xupinjie merged commit bdc58b0 into NVIDIA:main Mar 23, 2026
1 check was pending
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.

2 participants