.

huggingface · Feb 13, 2024 · 3bd346d · 3bd346d
1 parent b39c831
commit 3bd346d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/.github/workflows/small.yaml b/.github/workflows/small.yaml
@@ -56,4 +56,4 @@ jobs:
     - name: Run tests
       # NOTE: -m "not fa2" will run all the unit tests that don't have the mark
       # "fa2" (these are FA2-related tests, we can't run it on T4)
-      run: pytest -n 1 --color=yes --durations=0 --verbose tests/test_clip_grads
+      run: pytest -n 1 --color=yes --durations=0 --verbose tests/test_clip_grads.py::test_clip_grads_with_pp[inf]
diff --git a/tests/test_clip_grads.py b/tests/test_clip_grads.py
@@ -189,7 +189,8 @@ def _test_clip_grads_with_pp(parallel_context: ParallelContext, norm_type: float
             to_rank=reference_rank,
         )
 
-    # parallel_context.destroy()
+    parallel_context.destroyaa()
+    parallel_context.destroy()
 
 
 @pytest.mark.skipif(available_gpus() < 2, reason="test_clip_grads_with_tp requires at least 2 gpus")