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