From 7f124bee043527df3a92f8944edbb8521b9dfead Mon Sep 17 00:00:00 2001 From: Zhaolong Xing Date: Fri, 24 Apr 2020 16:30:40 +0800 Subject: [PATCH] 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 --- .../tests/unittests/ir/inference/test_trt_fc_fuse_pass.py | 3 ++- tools/manylinux1/Dockerfile.Inference | 6 +++--- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/python/paddle/fluid/tests/unittests/ir/inference/test_trt_fc_fuse_pass.py b/python/paddle/fluid/tests/unittests/ir/inference/test_trt_fc_fuse_pass.py index 9bd28cafd4c..48706bf5ad1 100644 --- a/python/paddle/fluid/tests/unittests/ir/inference/test_trt_fc_fuse_pass.py +++ b/python/paddle/fluid/tests/unittests/ir/inference/test_trt_fc_fuse_pass.py @@ -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, diff --git a/tools/manylinux1/Dockerfile.Inference b/tools/manylinux1/Dockerfile.Inference index 80c7787d8ab..8a72c3e0a36 100644 --- a/tools/manylinux1/Dockerfile.Inference +++ b/tools/manylinux1/Dockerfile.Inference @@ -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 -- GitLab