Adding some basic unit tests

4ec95a2e · shanmugamr · e9f20007 · 4ec95a2e
Commit 4ec95a2e authored Oct 06, 2022 by shanmugamr
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

.gitlab-ci.yml .gitlab-ci.yml +1 -1

No files found.
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -2,7 +2,7 @@ image: gitlab-master.nvidia.com/dl/dgx/pytorch:21.12-py3-devel

 test:
  tags:
-    - docker
+    - docker_gpu_enabled
  script:
    - nvidia-smi
    - torchrun --nproc_per_node=2  -m pytest --cov-report=term --cov-report=html --cov=megatron/core tests/