From e8d21b6349e74ae231b94435213d87a66193ad61 Mon Sep 17 00:00:00 2001 From: fengjiayi Date: Wed, 7 Mar 2018 18:47:08 +0800 Subject: [PATCH] fix an error --- paddle/fluid/operators/reader/CMakeLists.txt | 2 +- .../fluid/operators/reader/create_double_buffer_reader_op.cc | 4 +--- 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/paddle/fluid/operators/reader/CMakeLists.txt b/paddle/fluid/operators/reader/CMakeLists.txt index 4725de916f..335c5b26a8 100644 --- a/paddle/fluid/operators/reader/CMakeLists.txt +++ b/paddle/fluid/operators/reader/CMakeLists.txt @@ -2,5 +2,5 @@ cc_library(reader_op_registry SRCS reader_op_registry.cc DEPS operator op_regist op_library(create_random_data_generator_op SRCS create_random_data_generator_op.cc DEPS reader_op_registry) op_library(create_shuffle_reader_op SRCS create_shuffle_reader_op.cc DEPS reader_op_registry) op_library(create_batch_reader_op SRCS create_batch_reader_op.cc DEPS reader_op_registry) -op_library(create_double_buffer_reader_op SRCS create_double_buffer_reader_op.cc DEPS reader_op_registry threadpool) +op_library(create_double_buffer_reader_op SRCS create_double_buffer_reader_op.cc DEPS reader_op_registry) set(READER_LIBRARY create_random_data_generator_op create_shuffle_reader_op create_batch_reader_op create_double_buffer_reader_op PARENT_SCOPE) diff --git a/paddle/fluid/operators/reader/create_double_buffer_reader_op.cc b/paddle/fluid/operators/reader/create_double_buffer_reader_op.cc index 42aa9295a4..94f7f79789 100644 --- a/paddle/fluid/operators/reader/create_double_buffer_reader_op.cc +++ b/paddle/fluid/operators/reader/create_double_buffer_reader_op.cc @@ -12,7 +12,7 @@ // See the License for the specific language governing permissions and // limitations under the License. -#include "paddle/fluid/framework/threadpool.h" +#include #include "paddle/fluid/operators/reader/reader_op_registry.h" namespace paddle { @@ -31,8 +31,6 @@ class DoubleBufferReader : public framework::DecoratedReader { std::thread prefetch( std::bind(&DoubleBufferReader::PrefetchThreadFunc, this)); prefetch.detach(); - // framework::Async( - // std::bind(&DoubleBufferReader::PrefetchThreadFunc, this)); } void ReadNext(std::vector* out) override; -- GitLab