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

Co-Authored-By: Du Li <[email protected]>
dbyoung18 · Jun 11, 2024 · a58a3b3 · a58a3b3
1 parent ec4266b
commit a58a3b3
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