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