tests/pytorch/test_fused_attn.py · 325bf911cf37a7ff4aad2d116d560ec8e691e0dc · OpenDAS / TransformerEngine · GitLab

Find file Blame History Permalink

[PyTorch] Disable large test cases for Transformer layer (#508) · 325bf911
Tim Moon authored Nov 07, 2023
```
Disable large test cases for PyTorch Transformer layer
Signed-off-by: Tim Moon <tmoon@nvidia.com>
```
325bf911

test_fused_attn.py 37.5 KB