regisss HF staff commited on
Commit
bf9de03
1 Parent(s): 47d6637

Add cross_entropy to fp32 ops

Browse files
Files changed (1) hide show
  1. gaudi_config.json +2 -1
gaudi_config.json CHANGED
@@ -18,6 +18,7 @@
18
  "hmp_fp32_ops": [
19
  "embedding",
20
  "nll_loss",
21
- "log_softmax"
 
22
  ]
23
  }
 
18
  "hmp_fp32_ops": [
19
  "embedding",
20
  "nll_loss",
21
+ "log_softmax",
22
+ "cross_entropy"
23
  ]
24
  }