Update gaudi_config.json

#2
Files changed (1) hide show
  1. gaudi_config.json +27 -4
gaudi_config.json CHANGED
@@ -1,5 +1,28 @@
1
  {
2
- "use_torch_autocast": true,
3
- "use_fused_adam": true,
4
- "use_fused_clip_norm": true
5
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  {
2
+ "use_habana_mixed_precision": true,
3
+ "hmp_is_verbose": false,
4
+ "use_fused_adam": true,
5
+ "use_fused_clip_norm": true,
6
+ "hmp_bf16_ops": [
7
+ "add",
8
+ "addmm",
9
+ "bmm",
10
+ "div",
11
+ "dropout",
12
+ "gelu",
13
+ "iadd",
14
+ "linear",
15
+ "layer_norm",
16
+ "matmul",
17
+ "mm",
18
+ "rsub",
19
+ "softmax",
20
+ "truediv"
21
+ ],
22
+ "hmp_fp32_ops": [
23
+ "cross_entropy",
24
+ "embedding",
25
+ "nll_loss",
26
+ "log_softmax"
27
+ ]
28
+ }