- Notifications
You must be signed in to change notification settings - Fork202
Commite3e399a
authored
[5336870] AutoCast: Unblock LSTM from conversion (#544)
Blocking was added as a WAR for TensorRT bug 5336870, fixed in TensorRT10.14## What does this PR do?**Type of change:** Bug fix <!-- Use one of the following: Bug fix, newfeature, new example, new tests, documentation. -->**Overview:** Unblock LSTM from conversion to 16-bit in AutoCast.Blocking was added as a WAR for TensorRT bug 5336870 (TRT only supportedFP32 LSTM). Bug was fixed in TensorRT 10.14 which is now released.## Usage<!-- You can potentially add a usage example below. -->```python# Add a code snippet demonstrating how to use this```## Testing<!-- Mention how have you tested your change if applicable. -->## 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 [Contributorguidelines](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?**: No- **Did you add or update any necessary documentation?**: No- **Did you update[Changelog](https://github.com/NVIDIA/TensorRT-Model-Optimizer/blob/main/CHANGELOG.rst)?**:No <!--- Only for new features, API changes, critical bug fixes or bwbreaking changes. -->## Additional Information<!-- E.g. related issue. -->Signed-off-by: Gal Hubara Agam <96368689+galagam@users.noreply.github.com>1 parent4853668 commite3e399a
1 file changed
+1
-1
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
68 | 68 | | |
69 | 69 | | |
70 | 70 | | |
71 | | - | |
| 71 | + | |
72 | 72 | | |
73 | 73 | | |
74 | 74 | | |
| |||
0 commit comments
Comments
(0)