Skip to content

[Paddle Tensorrt] change int32 to int64 in test file #69191

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 14 commits into from
Dec 17, 2024
Merged

[Paddle Tensorrt] change int32 to int64 in test file #69191

Show file tree
Hide file tree
Changes from 8 commits
Commits
Show all changes
14 commits
Select commit Hold shift + click to select a range
79742c0
mod int64
Hanyonggong Nov 5, 2024
c1587fe
rm pass_test
Hanyonggong Nov 5, 2024
d4d9a09
change np
Hanyonggong Nov 5, 2024
9cc1137
change file
Hanyonggong Nov 5, 2024
790b729
fix bug
Hanyonggong Nov 12, 2024
7946c85
fix conflict
Hanyonggong Dec 5, 2024
8d71812
fix conflict
Hanyonggong Dec 5, 2024
69793e0
add test mod
Hanyonggong Dec 5, 2024
ff7e10a
Merge branch 'develop' of https://github.com/PaddlePaddle/Paddle into…
Hanyonggong Dec 6, 2024
9b5243d
Update
Hanyonggong Dec 6, 2024
0db85a0
fix conflict
Hanyonggong Dec 6, 2024
4890750
update third_party
Hanyonggong Dec 6, 2024
187f8d4
update flashattn
Hanyonggong Dec 6, 2024
dbd332e
change type
Hanyonggong Dec 13, 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
134 changes: 0 additions & 134 deletions test/tensorrt/pass_test.py
View file
Open in desktop

This file was deleted.

6 changes: 3 additions & 3 deletions test/tensorrt/test_converter_common.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -166,7 +166,7 @@ def setUp(self):
self.python_api = bilinear_python_api
self.api_args = {
"x": np.random.random([2, 3, 6, 10]).astype("float32"),
"OutSize": np.array([12, 12], dtype="int32"),
"OutSize": np.array([12, 12], dtype="int64"),
"SizeTensor": None,
"Scale": None,
"attrs": {
Expand Down Expand Up @@ -225,8 +225,8 @@ def setUp(self):
"x": x_nchw,
"OutSize": None,
"SizeTensor": [
np.array([12], dtype="int32"),
np.array([12], dtype="int32"),
np.array([12], dtype="int64"),
np.array([12], dtype="int64"),
],
"Scale": None,
"attrs": {
Expand Down
8 changes: 4 additions & 4 deletions test/tensorrt/test_converter_creation.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -104,9 +104,9 @@ class TestArangeTRTPattern(TensorRTBaseTest):
def setUp(self):
self.python_api = paddle.arange
self.api_args = {
"start": np.array([0]).astype("int32"),
"end": np.array([6]).astype("int32"),
"step": np.array([1]).astype("int32"),
"start": np.array([0]).astype("int64"),
"end": np.array([6]).astype("int64"),
"step": np.array([1]).astype("int64"),
}
self.program_config = {"feed_list": []}
self.min_shape = {}
Expand Down Expand Up @@ -195,7 +195,7 @@ class TestFullWithTensorTRTPattern(TensorRTBaseTest):
def setUp(self):
self.python_api = paddle.tensor.fill_constant
self.api_args = {
"shape": np.array([1]).astype("int32"),
"shape": np.array([1]).astype("int64"),
"dtype": "float32",
"value": np.array([0.0]).astype("float32"),
}
Expand Down
20 changes: 10 additions & 10 deletions test/tensorrt/test_converter_logic.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -24,8 +24,8 @@ class TestGreaterThanFloat32TRTPattern(TensorRTBaseTest):
def setUp(self):
self.python_api = paddle.greater_than
self.api_args = {
"x": np.random.randn(2, 3).astype(np.float32),
"y": np.random.randn(3).astype(np.float32),
"x": np.random.randn(2, 3).astype("float32"),
"y": np.random.randn(3).astype("float32"),
}
self.program_config = {"feed_list": ["x", "y"]}
self.min_shape = {"x": [1, 3], "y": [3]}
Expand All @@ -35,12 +35,12 @@ def test_trt_result(self):
self.check_trt_result()


class TestGreaterThanInt32TRTPattern(TensorRTBaseTest):
class TestGreaterThanInt64TRTPattern(TensorRTBaseTest):
def setUp(self):
self.python_api = paddle.greater_than
self.api_args = {
"x": np.random.randn(3).astype(np.int32),
"y": np.random.randn(3).astype(np.int32),
"x": np.random.randn(3).astype("int64"),
"y": np.random.randn(3).astype("int64"),
}
self.program_config = {"feed_list": ["x", "y"]}
self.min_shape = {"x": [1], "y": [1]}
Expand All @@ -54,8 +54,8 @@ class TestLessThanFloat32TRTPattern(TensorRTBaseTest):
def setUp(self):
self.python_api = paddle.less_than
self.api_args = {
"x": np.random.randn(2, 3).astype(np.float32),
"y": np.random.randn(3).astype(np.float32),
"x": np.random.randn(2, 3).astype("float32"),
Copy link
Contributor

@vivienfanghuagood vivienfanghuagood Dec 9, 2024

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

这类修改的作用是什么呢？

Copy link
Contributor Author

@Hanyonggong Hanyonggong Dec 9, 2024

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

规范了样式

"y": np.random.randn(3).astype("float32"),
}
self.program_config = {"feed_list": ["x", "y"]}
self.min_shape = {"x": [1, 3], "y": [3]}
Expand All @@ -65,12 +65,12 @@ def test_trt_result(self):
self.check_trt_result()


class TestLessThanInt32TRTPattern(TensorRTBaseTest):
class TestLessThanInt64TRTPattern(TensorRTBaseTest):
def setUp(self):
self.python_api = paddle.less_than
self.api_args = {
"x": np.random.randn(3).astype(np.int32),
"y": np.random.randn(3).astype(np.int32),
"x": np.random.randn(3).astype("int64"),
"y": np.random.randn(3).astype("int64"),
}
self.program_config = {"feed_list": ["x", "y"]}
self.min_shape = {"x": [1], "y": [1]}
Expand Down
Loading