From e51b6629e04075dc34ac008723ffbb599147b0fe Mon Sep 17 00:00:00 2001 From: Logan Adams Date: Tue, 26 Mar 2024 17:36:43 -0700 Subject: [PATCH] Correct typo in checking on bf16 unit test support --- tests/unit/checkpoint/test_lr_scheduler.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/unit/checkpoint/test_lr_scheduler.py b/tests/unit/checkpoint/test_lr_scheduler.py index 4891b4f6fa9b..89c4dd1b49f7 100644 --- 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