From 3f8b9dbc940723bdec47f333602ec9f8cf041e3d Mon Sep 17 00:00:00 2001 From: superjomn Date: Tue, 29 May 2018 18:57:27 +0800 Subject: [PATCH] update --- paddle/fluid/operators/CMakeLists.txt | 3 +-- paddle/fluid/operators/tensorrt_engine_op.cc | 1 - .../operators/tensorrt_engine_op_test.cc | 21 ------------------- 3 files changed, 1 insertion(+), 24 deletions(-) delete mode 100644 paddle/fluid/operators/tensorrt_engine_op_test.cc diff --git a/paddle/fluid/operators/CMakeLists.txt b/paddle/fluid/operators/CMakeLists.txt index 6e8f5aa52b8c5..e5c75cdb77fd9 100644 --- a/paddle/fluid/operators/CMakeLists.txt +++ b/paddle/fluid/operators/CMakeLists.txt @@ -221,7 +221,7 @@ op_library(cross_entropy_op DEPS cross_entropy) op_library(softmax_with_cross_entropy_op DEPS cross_entropy softmax) op_library(softmax_op DEPS softmax) op_library(sequence_softmax_op DEPS softmax) -if (TENSORRT_FOUND) +if (WITH_GPU && TENSORRT_FOUND) op_library(tensorrt_engine_op DEPS tensorrt_engine) endif() op_library(sum_op DEPS selected_rows_functor) @@ -300,4 +300,3 @@ cc_test(save_load_op_test SRCS save_load_op_test.cc DEPS save_op load_op) cc_test(save_load_combine_op_test SRCS save_load_combine_op_test.cc DEPS save_combine_op load_combine_op) nv_test(nccl_op_test SRCS nccl_op_test.cu.cc DEPS nccl_op gpu_info device_context) nv_test(dropout_op_test SRCS dropout_op_test.cc DEPS dropout_op tensor) -nv_test(tensorrt_engine_op_test SRCS tensorrt_engine_op_test.cc DEPS tensorrt_engine_op tensorrt_engine) diff --git a/paddle/fluid/operators/tensorrt_engine_op.cc b/paddle/fluid/operators/tensorrt_engine_op.cc index 7984be0fed62a..170e89ce9dae2 100644 --- a/paddle/fluid/operators/tensorrt_engine_op.cc +++ b/paddle/fluid/operators/tensorrt_engine_op.cc @@ -13,7 +13,6 @@ limitations under the License. */ #include "paddle/fluid/operators/tensorrt_engine_op.h" - #include "paddle/fluid/framework/op_registry.h" #include "paddle/fluid/inference/tensorrt/convert/op_converter.h" #include "paddle/fluid/inference/utils/singleton.h" diff --git a/paddle/fluid/operators/tensorrt_engine_op_test.cc b/paddle/fluid/operators/tensorrt_engine_op_test.cc deleted file mode 100644 index 715931ea04ccd..0000000000000 --- a/paddle/fluid/operators/tensorrt_engine_op_test.cc +++ /dev/null @@ -1,21 +0,0 @@ -/* Copyright (c) 2018 PaddlePaddle Authors. All Rights Reserved. - - Licensed under the Apache License, Version 2.0 (the "License"); - you may not use this file except in compliance with the License. - You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - - Unless required by applicable law or agreed to in writing, software - distributed under the License is distributed on an "AS IS" BASIS, - WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. - See the License for the specific language governing permissions and - limitations under the License. */ - -#include "paddle/fluid/operators/tensorrt_engine_op.h" -#include - -namespace paddle { - -namespace operators {} // namespace operators -} // namespace paddle