tests/jax/test_fused_attn.py · 9416519d8ce725e953a74e91fdb6cae32fb6516f · OpenDAS / TransformerEngine

"transformer_engine/jax/attention.py" did not exist on "b90a8d3a69ccc89c172ae9acc3925e45889ee578"

Kirthi Shankar Sivamani authored Jun 13, 2024



* Apply formatting
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

* Apply formatting
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

---------
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

9416519d

test_fused_attn.py 20 KB

Replace test_fused_attn.py