Skip to content

【快乐开源】Paddle Tensor 规范化二期：array_api_tests/test_creation_functions.py::test_arange鲁棒性增强 #71394

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 10 commits into from
Mar 6, 2025
Merged

【快乐开源】Paddle Tensor 规范化二期：array_api_tests/test_creation_functions.py::test_arange鲁棒性增强 #71394

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
10 commits
Select commit Hold shift + click to select a range
320f700
array_api_tests/test_creation_functions.py::test_arange鲁棒性增强
cszdrg Mar 4, 2025
5b5a8de
Merge branch 'develop' of https://github.com/cszdrg/Paddle into develop
cszdrg Mar 4, 2025
05cbb48
add test about 0 size output of arange in legacy_test/test_arange.py
cszdrg Mar 5, 2025
9997bfe
Merge branch 'PaddlePaddle:develop' into develop
cszdrg Mar 5, 2025
b15ab75
fix test about 0 size output of arange in legacy_test/test_arange.py
cszdrg Mar 5, 2025
79f99b9
fix test about 0 size output of arange in legacy_test/test_arange.py
cszdrg Mar 5, 2025
81701ca
fix test about 0 size output of arange in legacy_test/test_arange.py
cszdrg Mar 5, 2025
4358962
fix test about 0 size output of arange in legacy_test/test_arange.py
cszdrg Mar 5, 2025
3b406f6
fix test about 0 size output of arange in legacy_test/test_arange.py
cszdrg Mar 5, 2025
016a0e9
fix test about 0 size output of arange in legacy_test/test_arange.py
cszdrg Mar 5, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
17 changes: 8 additions & 9 deletions paddle/phi/kernels/funcs/range_function.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -26,18 +26,17 @@ void GetSize(T start, T end, T step, int64_t* size) {
common::errors::InvalidArgument("The step of range op should not be 0."));

if (start < end) {
PADDLE_ENFORCE_GT(
step,
0,
common::errors::InvalidArgument(
"The step should be greater than 0 while start < end."));
if (step < 0) {
*size = 0;
return;
}
}

if (start > end) {
PADDLE_ENFORCE_LT(step,
0,
common::errors::InvalidArgument(
"The step should be less than 0 while start > end."));
if (step > 0) {
*size = 0;
return;
}
}

*size = std::is_integral<T>::value
Expand Down
12 changes: 12 additions & 0 deletions test/legacy_test/test_arange.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -240,6 +240,18 @@ def test_out(self):
np.testing.assert_array_equal(x12.numpy(), x12_expected_data)
self.assertEqual(x12.numpy().dtype, np.int64)

# [start<end step<0]
x13 = paddle.arange(start=0, end=10, step=-1)

x13_expected_data = np.array([])
np.testing.assert_array_equal(x13.numpy(), x13_expected_data)

# [start>end step>0]
x14 = paddle.arange(start=10, end=0, step=1)

x14_expected_data = np.array([])
np.testing.assert_array_equal(x14.numpy(), x14_expected_data)

paddle.enable_static()


Expand Down