diff --git a/paddle/fluid/framework/tensor_util_test.cc b/paddle/fluid/framework/tensor_util_test.cc
index 994d88532bc19d602077513037b6caf4105cb00c..6e10885890cd2d4a0d77834944b37e291197b637 100644
--- a/paddle/fluid/framework/tensor_util_test.cc
+++ b/paddle/fluid/framework/tensor_util_test.cc
@@ -242,7 +242,9 @@ TEST(TensorContainsNAN, CPU) {
 
   {
     paddle::framework::Tensor src;
-    float16* buf = src.mutable_data<float16>({3}, paddle::platform::CPUPlace());
+    paddle::platform::float16* buf =
+        src.mutable_data<paddle::platform::float16>(
+            {3}, paddle::platform::CPUPlace());
     buf[0] = 0.0;
     buf[1].x = 0x7fff;
     buf[2] = 0.0;
@@ -266,7 +268,9 @@ TEST(TensorContainsInf, CPU) {
 
   {
     paddle::framework::Tensor src;
-    float16* buf = src.mutable_data<float16>({3}, paddle::platform::CPUPlace());
+    paddle::platform::float16* buf =
+        src.mutable_data<paddle::platform::float16>(
+            {3}, paddle::platform::CPUPlace());
     buf[0] = 1.0;
     buf[1].x = 0x7c00;
     buf[2] = 0.0;