Skip to content

Commit 90c5e2a

Browse files
committed
added tests to GPU tests CI
Signed-off-by: mikail <mkhona@nvidia.com>
1 parent e46ab2f commit 90c5e2a

1 file changed

Lines changed: 9 additions & 0 deletions

File tree

tests/ci/L0_Tests_GPU.sh

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,4 +13,13 @@
1313
# limitations under the License.
1414
CUDA_VISIBLE_DEVICES="0" coverage run -p --source=emerging_optimizers tests/test_muon_utils.py
1515
CUDA_VISIBLE_DEVICES="0" coverage run -p --source=emerging_optimizers tests/test_orthogonalized_optimizer.py
16+
CUDA_VISIBLE_DEVICES="0" coverage run -p --source=emerging_optimizers tests/test_soap_functions.py
17+
CUDA_VISIBLE_DEVICES="0" coverage run -p --source=emerging_optimizers tests/test_soap_utils.py
18+
CUDA_VISIBLE_DEVICES="0" coverage run -p --source=emerging_optimizers tests/soap_smoke_test.py
19+
CUDA_VISIBLE_DEVICES="0" coverage run -p --source=emerging_optimizers tests/soap_mnist_test.py
20+
CUDA_VISIBLE_DEVICES="0" coverage run -p --source=emerging_optimizers tests/test_scalar_optimizers.py
21+
22+
23+
24+
1625

0 commit comments

Comments
 (0)