From 1f93b60bd72cfd6e38a9f8be25545cee2dfb973b Mon Sep 17 00:00:00 2001 From: zyf654321 Date: Tue, 24 Sep 2024 17:37:27 +0800 Subject: [PATCH] limit only on cuda --- dipu/tests/python/unittests/test_adamw.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dipu/tests/python/unittests/test_adamw.py b/dipu/tests/python/unittests/test_adamw.py index c9cdced87..b04125653 100644 --- 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):