tests/tensorflow/test_transformer.py · 2574a1ca23f6d7fe9b4748c6cc347f158d232e22 · OpenDAS / TransformerEngine

Restrict TF tests to one GPU (#264) · 6b6823a1

Kaixi Hou authored Jun 08, 2023



* Only use one gpu for tensorflow tests
Signed-off-by: kaixih <kaixih@nvidia.com>

* Simplify the change
Signed-off-by: kaixih <kaixih@nvidia.com>

* Final fix
Signed-off-by: kaixih <kaixih@nvidia.com>

---------
Signed-off-by: kaixih <kaixih@nvidia.com>
Co-authored-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

6b6823a1

test_transformer.py 4.23 KB

Replace test_transformer.py