From 05857c6c891d84baa3f6d26eafa51a9c3bebcbc7 Mon Sep 17 00:00:00 2001
From: shuzhenyi <shuzhenyi@baidu.com>
Date: Sun, 14 Dec 2025 19:26:56 +0800
Subject: [PATCH] Fix grad_outputs placement in Paddle output grad generation

---
 tester/base.py | 1 +
 1 file changed, 1 insertion(+)

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":