Skip to content

Commit feb0ed1

Browse files
committed
fix paddle.NPUPlace
1 parent c93a7d6 commit feb0ed1

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

python/paddle/fluid/tests/unittests/npu/test_conv2d_transpose_op_npu.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -684,7 +684,7 @@ def error_groups():
684684

685685
class TestConv2DTransposeRepr(unittest.TestCase):
686686
def test_case(self):
687-
paddle.disable_static()
687+
paddle.disable_static(paddle.NPUPlace(0))
688688
x_var = paddle.uniform((2, 4, 8, 8), dtype='float32', min=-1., max=1.)
689689
conv = nn.Conv2DTranspose(4, 6, (3, 3), output_padding=1, stride=2)
690690
print(conv)

0 commit comments

Comments
 (0)