Unverified Commit 5412a334 authored by Shaden Smith's avatar Shaden Smith Committed by GitHub
Browse files

unit test rename (#442)

parent 6d176c45
......@@ -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))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment