From 6350528220e4cfb447d3496e388715c013574a40 Mon Sep 17 00:00:00 2001
From: Leo Chen <chenqiuliang@baidu.com>
Date: Tue, 23 Mar 2021 14:40:16 +0800
Subject: [PATCH] [NPU] support fp16 for npu accuracy op (#31797)

---
 paddle/fluid/operators/metrics/accuracy_op_npu.cc | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/paddle/fluid/operators/metrics/accuracy_op_npu.cc b/paddle/fluid/operators/metrics/accuracy_op_npu.cc
index 10b28b532b1..4ffcbaf5531 100644
--- a/paddle/fluid/operators/metrics/accuracy_op_npu.cc
+++ b/paddle/fluid/operators/metrics/accuracy_op_npu.cc
@@ -9,7 +9,6 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 See the License for the specific language governing permissions and
 limitations under the License. */
 
-#ifdef PADDLE_WITH_ASCEND_CL
 #include <memory>
 #include <string>
 
@@ -119,6 +118,7 @@ namespace ops = paddle::operators;
 
 REGISTER_OP_NPU_KERNEL(
     accuracy, ops::AccuracyNPUKernel<paddle::platform::NPUDeviceContext, float>,
+    ops::AccuracyNPUKernel<paddle::platform::NPUDeviceContext,
+                           paddle::platform::float16>,
     ops::AccuracyNPUKernel<paddle::platform::NPUDeviceContext, int>,
     ops::AccuracyNPUKernel<paddle::platform::NPUDeviceContext, int64_t>);
-#endif
-- 
GitLab