regisss HF Staff commited on
Commit
08d3597
·
1 Parent(s): abbc0a5

Update gaudi_config.json

Browse files
Files changed (1) hide show
  1. gaudi_config.json +22 -3
gaudi_config.json CHANGED
@@ -4,19 +4,31 @@
4
  "use_fused_adam": true,
5
  "use_fused_clip_norm": true,
6
  "hmp_bf16_ops": [
 
 
7
  "conv1d",
8
  "conv2d",
9
  "conv3d",
 
10
  "conv_transpose1d",
11
- "conv_transpose2d",
12
- "conv_transpose3d",
 
 
13
  "addmm",
 
 
14
  "matmul",
15
  "einsum",
16
  "mm",
17
  "mv",
 
18
  "linear",
 
 
19
  "bmm",
 
 
20
  "layer_norm",
21
  "group_norm"
22
  ],
@@ -35,12 +47,19 @@
35
  "rsqrt",
36
  "sinh",
37
  "tan",
38
- "pow",
 
 
39
  "softplus",
 
 
 
 
40
  "cosine_similarity",
41
  "poisson_nll_loss",
42
  "cosine_embedding_loss",
43
  "nll_loss",
 
44
  "hinge_embedding_loss",
45
  "kl_div",
46
  "l1_loss",
 
4
  "use_fused_adam": true,
5
  "use_fused_clip_norm": true,
6
  "hmp_bf16_ops": [
7
+ "_convolution.deprecated",
8
+ "_convolution",
9
  "conv1d",
10
  "conv2d",
11
  "conv3d",
12
+ "conv_tbc",
13
  "conv_transpose1d",
14
+ "conv_transpose2d.input",
15
+ "conv_transpose3d.input",
16
+ "convolution",
17
+ "prelu",
18
  "addmm",
19
+ "addmv",
20
+ "addr",
21
  "matmul",
22
  "einsum",
23
  "mm",
24
  "mv",
25
+ "silu",
26
  "linear",
27
+ "addbmm",
28
+ "baddbmm",
29
  "bmm",
30
+ "chain_matmul",
31
+ "linalg_multi_dot",
32
  "layer_norm",
33
  "group_norm"
34
  ],
 
47
  "rsqrt",
48
  "sinh",
49
  "tan",
50
+ "pow.Tensor_Scalar",
51
+ "pow.Tensor_Tensor",
52
+ "pow.Scalar",
53
  "softplus",
54
+ "frobenius_norm",
55
+ "frobenius_norm.dim",
56
+ "nuclear_norm",
57
+ "nuclear_norm.dim",
58
  "cosine_similarity",
59
  "poisson_nll_loss",
60
  "cosine_embedding_loss",
61
  "nll_loss",
62
+ "nll_loss2d",
63
  "hinge_embedding_loss",
64
  "kl_div",
65
  "l1_loss",