Skip to content

rewrite unittest of trt_activation_op#11222

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

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

Conversation

@luotao1
Copy link
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
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

Labels

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

2 participants

@luotao1 @Superjomn