Correct typo in checking on bf16 unit test support (#5317)

Co-Authored-By: Du Li <[email protected]>
microsoft · Apr 1, 2024 · 014576a · 014576a
1 parent ffb53c2
commit 014576a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/unit/checkpoint/test_lr_scheduler.py b/tests/unit/checkpoint/test_lr_scheduler.py
@@ -53,7 +53,7 @@ def test_checkpoint_lr_scheduler(self, tmpdir, zero_stage, use_cpu_offload):
         }
         if get_accelerator().is_fp16_supported():
             config_dict["fp16"] = {"enabled": True}
-        elif get_accelerator().is_fp16_supported():
+        elif get_accelerator().is_bf16_supported():
             config_dict["bf16"] = {"enabled": True}
         hidden_dim = 10