mirror of
https://github.com/NVIDIA/Model-Optimizer.git
synced 2026-11-08 09:03:20 +08:00
Riyad Islam
31c11329bd
## What does this PR do? **Type of change:** New feature <!-- Use one of the following: Bug fix, new feature, new example, new tests, documentation. --> **Overview:** Support quantization of PyTorch Geometric ```python # Add a code snippet demonstrating how to use this ``` ## Testing `python -m pytest tests/unit/torch/quantization/plugins/test_pytorch_geometric_plugin.py -v` ## 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 <!--- If No, explain why. --> - **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)?**: Yes<!--- Only for new features, API changes, critical bug fixes or bw breaking changes. --> ## Additional Information - https://github.com/NVIDIA/TensorRT-Model-Optimizer/issues/298 --------- Signed-off-by: Riyad Islam <rislam@nvidia.com>