deprecate is_torch_bf16_available (#17738)

* deprecate is_torch_bf16_available * address suggestions
2022-06-20 05:40:11 -07:00
parent 132402d752
commit a2d34b7c04
5 changed files with 47 additions and 19 deletions
--- a/tests/deepspeed/test_model_zoo.py
+++ b/tests/deepspeed/test_model_zoo.py
@@ -306,7 +306,7 @@ stages = [ZERO2, ZERO3]
 #
 # dtypes = [FP16]
 # so just hardcoding --fp16 for now
-# if is_torch_bf16_available():
+# if is_torch_bf16_gpu_available():
 #     dtypes += [BF16]