tests/pytorch/test_fused_optimizer.py · 23caab3fab07b212df65b002eeb05834e6f6c85e · OpenDAS / TransformerEngine · GitLab

Find file Blame History Permalink

[PyTorch] Port fused optimizer tests to pytest (#1185) · 7e1068b3
Tim Moon authored Sep 17, 2024
```
Port optimizer tests to pytest
Signed-off-by: Tim Moon <tmoon@nvidia.com>
```
7e1068b3

test_fused_optimizer.py 18.9 KB