transformer_engine/jax/cpp_extensions/attention.py · 4d65073fee2ddc9334bc8861832a8020a5618832 · OpenDAS / TransformerEngine · GitLab

Find file Blame History Permalink

[JAX] Fix for Disable FusedAttn with FFI by default (#1304) · 4b8ffef4
Phuong Nguyen authored Nov 01, 2024
```
rm default value for NVTE_JAX_FUSED_ATTN_WITH_FFI
Signed-off-by: Phuong Nguyen <phuonguyen@nvidia.com>
```
4b8ffef4

attention.py 63.6 KB