Add cross_entropy to fp32 ops
Browse files- 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 |
}
|