Skip to content

rewrite unittest of trt_activation_op#11222

Merged
luotao1 merged 4 commits into
PaddlePaddle:developfrom
luotao1:trt
Jun 7, 2018
Merged

rewrite unittest of trt_activation_op#11222
luotao1 merged 4 commits into
PaddlePaddle:developfrom
luotao1:trt

Conversation

@luotao1
Copy link
Copy Markdown
Contributor

@luotao1 luotao1 commented Jun 6, 2018

  • use UT_framework of tensorrt to rewrite unit test of trt_activation_op
  • add version check of tensorrt (version >= 4.0), since tensorrt 4.0 add support for the MatrixMultiply, which used in trt_mul_op.

@luotao1 luotao1 added the 预测 原名Inference，包含Capi预测问题等 label Jun 6, 2018
@luotao1 luotao1 requested a review from Superjomn June 6, 2018 13:06
Superjomn
Superjomn approved these changes Jun 7, 2018
View reviewed changes
Copy link
Copy Markdown
Contributor

@Superjomn Superjomn left a comment

Choose a reason for hiding this comment

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

LGTM

@luotao1 luotao1 merged commit 6d80dd5 into PaddlePaddle:develop Jun 7, 2018
@luotao1 luotao1 deleted the trt branch June 7, 2018 07:32
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

1 more reviewer

@Superjomn Superjomn Superjomn approved these changes

Reviewers whose approvals may not affect merge requirements

Assignees

No one assigned

Labels

预测 原名Inference，包含Capi预测问题等

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

2 participants

@luotao1 @Superjomn