tests/pytorch/test_onnx_export.py · bcbd4be0f3e03e055bd591804f4d6eded7155a58 · OpenDAS / TransformerEngine

"...targets/git@developer.sourcefind.cn:gaoqiong/migraphx.git" did not exist on "d705e4838d36db483ed80b55ce3977056360bb00"

Find file Blame History Permalink

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

test_onnx_export.py 38.2 KB

Replace test_onnx_export.py