Adding some basic unit tests

056fc7c9 · shanmugamr · 136cf036 · 056fc7c9 · 056fc7c9
Commit 056fc7c9 authored Oct 06, 2022 by shanmugamr
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

.coverage .coverage +0 -0

.gitlab-ci.yml .gitlab-ci.yml +2 -0

No files found.
--- a/.coverage
+++ b/.coverage
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -4,6 +4,8 @@ test:
  tags:
    - docker
  script:
+    - source /home/shanmugamr/test_env/bin/activate
+    - nvidia-smi
    - torchrun --nproc_per_node=2  -m pytest --cov-report=term --cov-report=html --cov=megatron/core tests/
  artifacts:
    paths: