From 9c8189aac562e9cb3dd4c195a3ebbe7e72746897 Mon Sep 17 00:00:00 2001 From: Shaden Smith Date: Thu, 24 Sep 2020 14:25:13 -0700 Subject: [PATCH] unit test rename --- 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 d1d03f500af9..4cea20d4e250 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))