Intel Gen 4 Xeon and later support bf16 (#6367)

jiqing-feng · web-flow · commit f2d51a28f717 · 2024-01-05T11:47:28.000+05:30
* Intel Gen 4 Xeon and later support bf16

* fix bf16 notes
diff --git a/examples/textual_inversion/textual_inversion.py b/examples/textual_inversion/textual_inversion.py
@@ -341,7 +341,7 @@ def parse_args():
         help=(
             "Whether to use mixed precision. Choose"
             "between fp16 and bf16 (bfloat16). Bf16 requires PyTorch >= 1.10."
-            "and an Nvidia Ampere GPU."
+            "and Nvidia Ampere GPU or Intel Gen 4 Xeon (and later) ."
         ),
     )
     parser.add_argument(

Original file line number	Diff line number	Diff line change
`@@ -341,7 +341,7 @@ def parse_args():`
`341`	`341`	`help=(`
`342`	`342`	`"Whether to use mixed precision. Choose"`
`343`	`343`	`"between fp16 and bf16 (bfloat16). Bf16 requires PyTorch >= 1.10."`
`344`		`- "and an Nvidia Ampere GPU."`
	`344`	`+ "and Nvidia Ampere GPU or Intel Gen 4 Xeon (and later) ."`
`345`	`345`	`),`
`346`	`346`	`)`
`347`	`347`	`parser.add_argument(`