tests/pytorch/test_sanity.py · bacefdbb6815159c42d5ca501a1400697b98a1e3 · OpenDAS / TransformerEngine

[PyTorch] Reduce size of sanity tests (#510) · e2a75314

Tim Moon authored Jan 05, 2024



* Reduce size of PyT sanity tests
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Add test cases with 126M model and weird dimensions
Signed-off-by: Tim Moon <tmoon@nvidia.com>

* Add missing arg in GPT 126M test
Signed-off-by: Tim Moon <tmoon@nvidia.com>

---------
Signed-off-by: Tim Moon <tmoon@nvidia.com>
Signed-off-by: Tim Moon <4406448+timmoon10@users.noreply.github.com>

e2a75314

test_sanity.py 28 KB

Replace test_sanity.py