transformer_engine/jax/cpp_extensions.py · 90c267f2748ef4b4cee2ef3ef3068f5e91ed645b · OpenDAS / TransformerEngine · GitLab

Find file Blame History Permalink

[JAX] [B] Fixed Batcher in DBiasCastTranspose Primitive (#843) · 90c267f2
Phuong Nguyen authored May 13, 2024
```
fixed batcher in dbias_cast_transpose primitive
Signed-off-by: Phuong Nguyen <phuonguyen@nvidia.com>
```
90c267f2

cpp_extensions.py 184 KB