regisss HF staff commited on
Commit
d085d89
1 Parent(s): 33e34c0

Add cross_entropy to fp32 ops

Browse files
Files changed (1) hide show
  1. gaudi_config.json +2 -1
gaudi_config.json CHANGED
@@ -23,6 +23,7 @@
23
  "hmp_fp32_ops": [
24
  "embedding",
25
  "nll_loss",
26
- "log_softmax"
 
27
  ]
28
  }
23
  "hmp_fp32_ops": [
24
  "embedding",
25
  "nll_loss",
26
+ "log_softmax",
27
+ "cross_entropy"
28
  ]
29
  }