regisss HF staff commited on
Commit
2f59071
1 Parent(s): 0d373f2

Replace hmp_*_ops by autocast_*_ops

Browse files
Files changed (1) hide show
  1. 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
- "hmp_bf16_ops": [
6
  "_convolution.deprecated",
7
  "_convolution",
8
  "conv1d",
@@ -31,7 +31,7 @@
31
  "layer_norm",
32
  "group_norm"
33
  ],
34
- "hmp_fp32_ops": [
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",