未验证 提交 7f124bee 编写于 作者: Z Zhaolong Xing 提交者: GitHub

cherry-pick: add trt check for converage ci (#23738) (#24096)

* cherry-pick: fix converage ci (#23738)

test=release/2.0-beta

* modity ut
test=release/2.0-beta
上级 87694b12
......@@ -25,7 +25,8 @@ from paddle.fluid.core import AnalysisConfig
class FCFusePassTRTTest(InferencePassTest):
def setUp(self):
with fluid.program_guard(self.main_program, self.startup_program):
data = fluid.data(name="data", shape=[32, 128], dtype="float32")
data = fluid.data(
name="data", shape=[32, 128, 2, 2], dtype="float32")
fc_out1 = fluid.layers.fc(input=data,
size=128,
num_flatten_dims=1,
......
......@@ -130,9 +130,9 @@ RUN curl -s -q https://glide.sh/get | sh
# 2. Manually add ~IPluginFactory() in IPluginFactory class of NvInfer.h, otherwise, it couldn't work in paddle.
# See https://github.com/PaddlePaddle/Paddle/issues/10129 for details.
RUN wget -q https://paddle-ci.gz.bcebos.com/TensorRT-5.1.5.0.Ubuntu-14.04.5.x86_64-gnu.cuda-10.1.cudnn7.5.tar.gz --no-check-certificate && \
tar -zxf TensorRT-5.1.5.0.Ubuntu-14.04.5.x86_64-gnu.cuda-10.1.cudnn7.5.tar.gz -C /usr/local && \
cp -rf /usr/local/TensorRT-5.1.5.0/include /usr
RUN wget -q https://paddlepaddledeps.cdn.bcebos.com/TensorRT-6.0.1.5.Ubuntu-16.04.x86_64-gnu.cuda-10.1.cudnn7.6.tar.gz --no-check-certificate && \
tar -zxf TensorRT-6.0.1.5.Ubuntu-16.04.x86_64-gnu.cuda-10.1.cudnn7.6.tar.gz -C /usr/local && \
cp -rf /usr/local/TensorRT-6.0.1.5/include/* /usr/include/ && cp -rf /usr/local/TensorRT-6.0.1.5/lib/* /usr/lib/
# git credential to skip password typing
RUN git config --global credential.helper store
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册