tests/tensorflow/test_layers.py · 38b85c356caf2c5972b88b80ba02d957dac69a8d · OpenDAS / TransformerEngine

"vscode:/vscode.git/clone" did not exist on "e9d3f2ff7772c8efe41dc805cec71c223ec18ec8"

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_layers.py 21.7 KB

Replace test_layers.py