Skip to content

Conversation

@baoachun
Copy link
Contributor

@baoachun baoachun commented Sep 7, 2021

PR types

Others

PR changes

Others

Describe

Add gather trt converter test case.

@paddle-bot-old
Copy link

paddle-bot-old bot commented Sep 7, 2021

Thanks for your contribution!
Please wait for the result of CI firstly. See Paddle CI Manual for details.


for shape in [[32], [16, 64], [32, 16, 16], [32, 64, 16, 32]]:
for index in [[1, 4], [4, 8]]:
for axis in [0, 1, 2, 3]:
Copy link
Contributor

Choose a reason for hiding this comment

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

这个axis也不支持负数吧

Copy link
Contributor Author

Choose a reason for hiding this comment

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

测试发现置为-1时原生和trt都能跑,设为其他负数会显示错误的显存申请数而报错。

Copy link
Contributor

@jiweibo jiweibo left a comment

Choose a reason for hiding this comment

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

LGTM

@jiweibo jiweibo merged commit 75d5e3b into PaddlePaddle:develop Sep 13, 2021
AnnaTrainingG pushed a commit to AnnaTrainingG/Paddle that referenced this pull request Sep 29, 2021
@baoachun baoachun deleted the gather_trt_check branch December 31, 2021 07:27
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants