fix test_custom_relu_model

PaddlePaddle · Apr 25, 2021 · 16c1564 · 16c1564 · paddle-bot-old · Apr 26, 2021
1 parent 968d13d
commit 16c1564
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/paddle/fluid/tests/custom_op/test_custom_relu_model.py b/python/paddle/fluid/tests/custom_op/test_custom_relu_model.py
@@ -247,7 +247,8 @@ def train_model(self, device, use_custom_op=False, use_pe=False):
         paddle.set_device(device)
 
         with paddle.static.scope_guard(paddle.static.Scope()):
-            with paddle.static.program_guard(paddle.static.Program()):
+            with paddle.static.program_guard(paddle.static.Program(),
+                                             paddle.static.Program()):
                 x = paddle.static.data(
                     shape=[None, self.in_dim], name='x', dtype='float32')
                 y = paddle.static.data(