From 384cd263143476dd113b9aa71897e4331742c4a5 Mon Sep 17 00:00:00 2001 From: HELSON Date: Wed, 30 Nov 2022 12:09:32 +0800 Subject: [PATCH] [zero] fix testing parameters (#2042) --- tests/test_gemini/update/test_optim.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_gemini/update/test_optim.py b/tests/test_gemini/update/test_optim.py index 89b9b433b..cd2d7155f 100644 --- a/tests/test_gemini/update/test_optim.py +++ b/tests/test_gemini/update/test_optim.py @@ -44,7 +44,7 @@ TEST_MODELS = ['gpt2', 'bert'] EXAMPLE_MODELS = ['simple_net'] -@parameterize('placement_policy', ['cuda']) +@parameterize('placement_policy', ['cuda', 'cpu', 'auto', 'const']) @parameterize('model_name', TEST_MODELS) def exam_model_step(placement_policy, model_name: str): set_seed(42)