diff --git a/python/paddle/fluid/tests/unittests/test_cross_entropy_loss.py b/python/paddle/fluid/tests/unittests/test_cross_entropy_loss.py
index 3ed8a69475c6499dcd95d98e790e569f199e3076..91237eb6d677faf5e441d2d9f835cf492fef1127 100644
--- a/python/paddle/fluid/tests/unittests/test_cross_entropy_loss.py
+++ b/python/paddle/fluid/tests/unittests/test_cross_entropy_loss.py
@@ -1533,9 +1533,9 @@ class TestCrossEntropyFAPIError(unittest.TestCase):
             self.assertRaises(ValueError, test_LabelValue_ExceedMin)
 
             def static_test_WeightLength_NotEqual():
-                input_np = np.random.random([2, 4]).astype(self.dtype)
+                input_np = np.random.random([2, 4]).astype('float32')
                 label_np = np.random.randint(0, 4, size=(2)).astype(np.int64)
-                weight_np = np.random.random([3]).astype(self.dtype)  #shape:C
+                weight_np = np.random.random([3]).astype('float32')
                 paddle.enable_static()
                 prog = fluid.Program()
                 startup_prog = fluid.Program()
@@ -1543,11 +1543,11 @@ class TestCrossEntropyFAPIError(unittest.TestCase):
                 ) else fluid.CPUPlace()
                 with fluid.program_guard(prog, startup_prog):
                     input = fluid.data(
-                        name='input', shape=[2, 4], dtype=self.dtype)
+                        name='input', shape=[2, 4], dtype='float32')
                     label = fluid.data(name='label', shape=[2], dtype='int64')
                     weight = fluid.data(
                         name='weight', shape=[3],
-                        dtype=self.dtype)  #weight for each class
+                        dtype='float32')  #weight for each class
                     cross_entropy_loss = paddle.nn.loss.CrossEntropyLoss(
                         weight=weight)
                     ret = cross_entropy_loss(input, label)