[test] fix unit test (#31)

delock · Aug 9, 2022 · ec9c7a3 · ec9c7a3
1 parent 3967718
commit ec9c7a3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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