Skip to content

[PIR] D-23 Adapt randperm test_errors #62797

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
0x45f merged 7 commits into from
Mar 19, 2024
Merged

[PIR] D-23 Adapt randperm test_errors #62797

Show file tree
Hide file tree
Changes from 6 commits
Commits
Show all changes
7 commits
Select commit Hold shift + click to select a range
c638c4e
test_errors_d_23
ooooo-create Mar 17, 2024
8011f6b
fix
ooooo-create Mar 17, 2024
83102af
fix ValueError
ooooo-create Mar 17, 2024
14ab72c
fix ValueError
ooooo-create Mar 17, 2024
5ff4281
fix ValueError
ooooo-create Mar 17, 2024
c13c74c
fix ge to gt
ooooo-create Mar 17, 2024
9296fce
skip a case in pir_mode
ooooo-create Mar 18, 2024
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
4 changes: 4 additions & 0 deletions paddle/phi/infermeta/nullary.cc
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -136,6 +136,10 @@ void PartialRecvInferMeta(int ring_id,
}

void RandpermInferMeta(int n, DataType dtype, MetaTensor* out) {
PADDLE_ENFORCE_GT(
n,
0,
errors::InvalidArgument("The upper bound %d isn't greater than 0.", n));
out->set_dims(common::make_ddim({n}));
out->set_dtype(dtype);
}
Expand Down
3 changes: 3 additions & 0 deletions test/legacy_test/test_randperm_op.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,7 @@

import paddle
from paddle.base import core
from paddle.pir_utils import test_with_pir_api
from paddle.static import Program, program_guard


Expand Down Expand Up @@ -156,13 +157,15 @@ def verify_output(self, outs):


class TestRandpermOpError(unittest.TestCase):
@test_with_pir_api
def test_errors(self):
with program_guard(Program(), Program()):
self.assertRaises(ValueError, paddle.randperm, -3)
self.assertRaises(TypeError, paddle.randperm, 10, 'int8')


class TestRandpermAPI(unittest.TestCase):
@test_with_pir_api
def test_out(self):
n = 10
place = (
Expand Down