From ec9c7a31b7079b4cd0b7d12ae208f8c78152b78c Mon Sep 17 00:00:00 2001 From: "Ma, Guokai" Date: Wed, 8 Jun 2022 13:53:28 +0800 Subject: [PATCH] [test] fix unit test (#31) --- deepspeed/runtime/fp16/unfused_optimizer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deepspeed/runtime/fp16/unfused_optimizer.py b/deepspeed/runtime/fp16/unfused_optimizer.py index fe8ba68da532..e241f33e51ec 100755 --- a/deepspeed/runtime/fp16/unfused_optimizer.py +++ b/deepspeed/runtime/fp16/unfused_optimizer.py @@ -44,7 +44,7 @@ def __init__(self, if dist.get_rank() == 0: logger.info(f'Fused Lamb Legacy : {self.fused_lamb_legacy} ') - if not accel_runtime.is_fp16_supported(): + if not accel_runtime.is_available(): raise SystemError("No accelerator or accelerator does not support FP16.") self.optimizer = init_optimizer