Skip to content

Commit e8d21b6

Browse files
committed
fix an error
1 parent 4fb7b96 commit e8d21b6

File tree

2 files changed

+2
-4
lines changed

2 files changed

+2
-4
lines changed

paddle/fluid/operators/reader/CMakeLists.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,5 +2,5 @@ cc_library(reader_op_registry SRCS reader_op_registry.cc DEPS operator op_regist
22
op_library(create_random_data_generator_op SRCS create_random_data_generator_op.cc DEPS reader_op_registry)
33
op_library(create_shuffle_reader_op SRCS create_shuffle_reader_op.cc DEPS reader_op_registry)
44
op_library(create_batch_reader_op SRCS create_batch_reader_op.cc DEPS reader_op_registry)
5-
op_library(create_double_buffer_reader_op SRCS create_double_buffer_reader_op.cc DEPS reader_op_registry threadpool)
5+
op_library(create_double_buffer_reader_op SRCS create_double_buffer_reader_op.cc DEPS reader_op_registry)
66
set(READER_LIBRARY create_random_data_generator_op create_shuffle_reader_op create_batch_reader_op create_double_buffer_reader_op PARENT_SCOPE)

paddle/fluid/operators/reader/create_double_buffer_reader_op.cc

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
// See the License for the specific language governing permissions and
1313
// limitations under the License.
1414

15-
#include "paddle/fluid/framework/threadpool.h"
15+
#include <thread>
1616
#include "paddle/fluid/operators/reader/reader_op_registry.h"
1717

1818
namespace paddle {
@@ -31,8 +31,6 @@ class DoubleBufferReader : public framework::DecoratedReader {
3131
std::thread prefetch(
3232
std::bind(&DoubleBufferReader::PrefetchThreadFunc, this));
3333
prefetch.detach();
34-
// framework::Async(
35-
// std::bind(&DoubleBufferReader::PrefetchThreadFunc, this));
3634
}
3735

3836
void ReadNext(std::vector<framework::LoDTensor>* out) override;

0 commit comments

Comments
 (0)