Skip to content

mlir-tensorrt-v0.1.32

Latest
Latest
Compare
Choose a tag to compare
@christopherbate christopherbate released this 23 Aug 18:00
· 54 commits to main since this release
mlir-tensorrt-v0.1.32
0a6a4dc
This commit was created on GitHub.com and signed with GitHub’s verified signature.
GPG key ID: B5690EEEBB952194
Learn about vigilant mode. 
Make `tp.Layernorm` 1:1 with torch (#119)

- Make the `normalized_shape` argument 1:1 with Torch. Previously, our
API only supported taking a single integer for the normalization
dimension. Now, a list of integers can be provided.
- Add integration tests for `tp.LayerNorm` and `tp.GroupNorm`
Assets 12
Loading