limit only on cuda

DeepLink-org · Sep 24, 2024 · cb2e125 · cb2e125
1 parent ffcc77f
commit cb2e125
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/dipu/tests/python/unittests/test_adamw.py b/dipu/tests/python/unittests/test_adamw.py
@@ -189,9 +189,11 @@ def adamw_(self, dtype_):
                 ),
             )
 
+    @onlyOn("CUDA")
     def test_adamw_fp16_(self):
         self.adamw_(torch.float16)
 
+    @onlyOn("CUDA")
     def test_adamw_fp32_(self):
         self.adamw_(torch.float32)