Skip to content

Commit a56893e

Browse files
change rand param name
1 parent faa8018 commit a56893e

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

test/legacy_test/test_set_value_op.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1794,16 +1794,16 @@ def test_value_input_is_scalar(self):
17941794
)
17951795
class TestSetValueWithStrideError(unittest.TestCase):
17961796
def test_same_place(self):
1797-
x = paddle.rand([5, 10], device=paddle.CUDAPlace(0))
1798-
y = paddle.rand([10, 5], device=paddle.CUDAPlace(0))
1797+
x = paddle.rand([5, 10], place=paddle.CUDAPlace(0))
1798+
y = paddle.rand([10, 5], place=paddle.CUDAPlace(0))
17991799
y.transpose_([1, 0])
18001800
x.set_value(y)
18011801
assert x.is_contiguous()
18021802

18031803
def test_different_place1(self):
18041804
# src place != dst place && src is not contiguous
1805-
x = paddle.rand([5, 10], device=paddle.CUDAPlace(0))
1806-
y = paddle.rand([10, 5], device=paddle.CPUPlace())
1805+
x = paddle.rand([5, 10], place=paddle.CUDAPlace(0))
1806+
y = paddle.rand([10, 5], place=paddle.CPUPlace())
18071807
y.transpose_([1, 0])
18081808
x.set_value(y)
18091809
assert not x.is_contiguous()
@@ -1813,7 +1813,7 @@ def test_different_place2(self):
18131813
with self.assertRaises(SystemError):
18141814
x = paddle.ones([5, 4], device=paddle.CUDAPlace(0))
18151815
x.transpose_([1, 0])
1816-
y = paddle.rand([4, 2], device=paddle.CPUPlace())
1816+
y = paddle.rand([4, 2], place=paddle.CPUPlace())
18171817
assert not x.is_contiguous()
18181818

18191819
x[:, 1:3].set_value(y)

0 commit comments

Comments
 (0)