Habana
regisss HF staff commited on
Commit
bf9d59b
·
1 Parent(s): c4587e6

Add cross_entropy to fp32 ops

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