Replace hmp_*_ops by autocast_*_ops
Browse files- gaudi_config.json +2 -2
gaudi_config.json
CHANGED
@@ -2,7 +2,7 @@
|
|
2 |
"use_torch_autocast": true,
|
3 |
"use_fused_adam": true,
|
4 |
"use_fused_clip_norm": true,
|
5 |
-
"
|
6 |
"_convolution.deprecated",
|
7 |
"_convolution",
|
8 |
"conv1d",
|
@@ -31,7 +31,7 @@
|
|
31 |
"layer_norm",
|
32 |
"group_norm"
|
33 |
],
|
34 |
-
"
|
35 |
"acos",
|
36 |
"asin",
|
37 |
"cosh",
|
|
|
2 |
"use_torch_autocast": true,
|
3 |
"use_fused_adam": true,
|
4 |
"use_fused_clip_norm": true,
|
5 |
+
"autocast_bf16_ops": [
|
6 |
"_convolution.deprecated",
|
7 |
"_convolution",
|
8 |
"conv1d",
|
|
|
31 |
"layer_norm",
|
32 |
"group_norm"
|
33 |
],
|
34 |
+
"autocast_fp32_ops": [
|
35 |
"acos",
|
36 |
"asin",
|
37 |
"cosh",
|