regisss HF staff commited on
Commit
0ec557c
1 Parent(s): 5b05242

Add cross_entropy to fp32 ops

Browse files
Files changed (1) hide show
  1. gaudi_config.json +3 -1
gaudi_config.json CHANGED
@@ -22,5 +22,7 @@
22
  "rsub",
23
  "softmax"
24
  ],
25
- "hmp_fp32_ops": []
 
 
26
  }
 
22
  "rsub",
23
  "softmax"
24
  ],
25
+ "hmp_fp32_ops": [
26
+ "cross_entropy"
27
+ ]
28
  }