limit only on cuda

DeepLink-org · Sep 24, 2024 · 1f93b60 · 1f93b60
1 parent cb2e125
commit 1f93b60
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dipu/tests/python/unittests/test_adamw.py b/dipu/tests/python/unittests/test_adamw.py
@@ -1,6 +1,6 @@
 import torch
 import numpy as np
-from torch_dipu.testing._internal.common_utils import TestCase, run_tests
+from torch_dipu.testing._internal.common_utils import TestCase, run_tests, onlyOn
 
 
 class TestFusedAdamW(TestCase):