tests/pytorch/test_fused_attn.py · 696ad6c4df75d8ecda6d3e8eb017ba63c75e939f · OpenDAS / TransformerEngine · GitLab

Find file Blame History Permalink

[Common/PyTorch] Fix FP8 fused attention input args (#592) · 696ad6c4
cyanguwa authored Jan 07, 2024
```
fix FP8 dims
Signed-off-by: Charlene Yang <8636796+cyanguwa@users.noreply.github.com>
```
696ad6c4

test_fused_attn.py 51.6 KB