Test

gaudi_config.json

@@ -23,6 +23,7 @@
   "hmp_fp32_ops": [
     "embedding",
     "nll_loss",
-    "log_softmax"
+    "log_softmax",
+    "cross_entropy"
   ]
 }