[fp16] fix a mapping error that would cause unit test to fail

delock · Aug 9, 2022 · 3967718 · 3967718
1 parent 961b2b4
commit 3967718
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepspeed/runtime/fp16/fused_optimizer.py b/deepspeed/runtime/fp16/fused_optimizer.py
@@ -44,7 +44,7 @@ def __init__(self,
         self.deepspeed = deepspeed
         self.has_moe_layers = has_moe_layers
         self.using_pipeline = self.deepspeed.pipeline_parallelism
-        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