From 5412a3341fa5d0211629ee87899015f98a62e0cc Mon Sep 17 00:00:00 2001 From: Shaden Smith Date: Fri, 25 Sep 2020 08:17:44 -0700 Subject: [PATCH] unit test rename (#442) --- tests/unit/{test_adam_acuracy.py => test_cpu_adam.py} | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) rename tests/unit/{test_adam_acuracy.py => test_cpu_adam.py} (94%) diff --git a/tests/unit/test_adam_acuracy.py b/tests/unit/test_cpu_adam.py similarity index 94% rename from tests/unit/test_adam_acuracy.py rename to tests/unit/test_cpu_adam.py index d1d03f50..4cea20d4 100755 --- a/tests/unit/test_adam_acuracy.py +++ b/tests/unit/test_cpu_adam.py @@ -31,7 +31,7 @@ def check_equal(first, second, atol=1e-2, verbose=False): (1024), (1048576), ]) # yapf: disable -def test_adam_opt(model_size): +def test_cpu_adam_opt(model_size): device = 'cpu' rng_state = torch.get_rng_state() param = torch.nn.Parameter(torch.randn(model_size, device=device)) -- GitLab