diff --git a/tester/base.py b/tester/base.py
index 27d9fb5d..ffac087a 100644
--- a/tester/base.py
+++ b/tester/base.py
@@ -610,6 +610,7 @@ def gen_paddle_output_and_output_grad(self, outputs):
             result_output_grad = paddle.to_tensor(
                 numpy_tensor,
                 dtype=dtype if dtype != "bfloat16" else "float32",
+                place=result_outputs[i].place,
             )
             result_output_grad.stop_gradient = False
             if dtype == "bfloat16":