mirror of
https://github.com/NVIDIA/Model-Optimizer.git
synced 2026-11-09 09:14:17 +08:00
Gal Hubara-Agam
ae915eaea9
## What does this PR do? new feature **Overview:** AutoCast: Allow configuring target opset from CLI and main API. ## Usage <!-- You can potentially add a usage example below. --> python -m modelopt.onnx.autocast --onnx_path model.onnx --opset 22 ```python # Add a code snippet demonstrating how to use this ``` ## Testing <!-- Mention how have you tested your change if applicable. --> pytest tests/unit/onnx/autocast/test_autocast.py -k test_opset ## Before your PR is "*Ready for review*" <!-- If you haven't finished some of the above items you can still open `Draft` PR. --> - **Make sure you read and follow [Contributor guidelines](https://github.com/NVIDIA/TensorRT-Model-Optimizer/blob/main/CONTRIBUTING.md)** and your commits are signed. - **Is this change backward compatible?**: Yes - **Did you write any new necessary tests?**: Yes - **Did you add or update any necessary documentation?**: Yes - **Did you update [Changelog](https://github.com/NVIDIA/TensorRT-Model-Optimizer/blob/main/CHANGELOG.rst)?**: No - minor API addition <!--- Only for new features, API changes, critical bug fixes or bw breaking changes. --> ## Additional Information <!-- E.g. related issue. --> Signed-off-by: Gal Hubara Agam <96368689+galagam@users.noreply.github.com>