Merge branch 'vitalyk/testfix' into 'main'

Remove CUDA requirement from cpu test. See merge request ADLR/megatron-lm!2199
Baibaifan · Oct 9, 2024 · 228dc20 · 228dc20
2 parents fcdbf90 + 37a2116
commit 228dc20
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/tests/unit_tests/transformer/test_rope.py b/tests/unit_tests/transformer/test_rope.py
@@ -43,7 +43,6 @@ def test_gpu_forward(self):
         assert output.dtype == torch.float32
         assert output.device.type == 'cuda'
 
-    @pytest.mark.skipif(not torch.cuda.is_available(), reason="CUDA not available")
     def test_cpu_forward(self):
         output = self.rope_cpu_init(64)
         assert output.shape[0] == 64