-
Notifications
You must be signed in to change notification settings - Fork 5.9k
[Paddle Inference]Add scale TRT converter unittest. #35225
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.
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
[Paddle Inference]Add scale TRT converter unittest. #35225
Conversation
|
Thanks for your contribution! |
|
|
||
| class TrtConvertScaleTest(TrtLayerAutoScanTest): | ||
| def is_program_valid(self, program_config: ProgramConfig) -> bool: | ||
| inputs = program_config.inputs |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
如果program合理的话,直接返回True即可
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Done
| program_config.ops[i].attrs | ||
| for i in range(len(program_config.ops)) | ||
| ] | ||
| if self.dims == 1: |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
这里是指dims==1的时候,动态shape支持,静态shape不支持吗?
|
|
||
| self.add_skip_case( | ||
| teller1, SkipReasons.TRT_NOT_SUPPORT, | ||
| "INPUT ScaleTensor and Shape NOT SUPPORT: we need to add support in the future" |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
如果是日后需要支持的feature的话,请使用TRT_NOT_IMPLEMENTED
如果是trt本身不支持的,使用TRT_NOT_SUPPORT
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Done
cryoco
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
* add_scale_teller * add_scale_teller * add_scale_teller * add_scale_teller * add_scale_teller * add_scale_teller * add_scale_teller * add_scale_teller * add_scale_teller * add_scale_teller * add_scale_teller * add_scale_teller * add_scale_teller * add_scale_teller
PR types
Function optimization
PR changes
OPs
Describe
增加op_converter中scale_op的单测