transformer_engine/pytorch/export.py · 156a074a979efef9220763537d324217816691e0 · OpenDAS / TransformerEngine

Fix FlashAttention tests (#99) · bcbd4be0

tcherckez-nvidia authored Mar 29, 2023


Signed-off-by: Tal Cherckez <tcherckez@nvidia.com>
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
Co-authored-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

bcbd4be0

export.py 918 Bytes

Replace export.py