File tree Expand file tree Collapse file tree 2 files changed +2
-4
lines changed
paddle/fluid/operators/reader Expand file tree Collapse file tree 2 files changed +2
-4
lines changed Original file line number Diff line number Diff line change @@ -2,5 +2,5 @@ cc_library(reader_op_registry SRCS reader_op_registry.cc DEPS operator op_regist
2
2
op_library (create_random_data_generator_op SRCS create_random_data_generator_op.cc DEPS reader_op_registry )
3
3
op_library (create_shuffle_reader_op SRCS create_shuffle_reader_op.cc DEPS reader_op_registry )
4
4
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 )
6
6
set (READER_LIBRARY create_random_data_generator_op create_shuffle_reader_op create_batch_reader_op create_double_buffer_reader_op PARENT_SCOPE )
Original file line number Diff line number Diff line change 12
12
// See the License for the specific language governing permissions and
13
13
// limitations under the License.
14
14
15
- #include " paddle/fluid/framework/threadpool.h "
15
+ #include < thread >
16
16
#include " paddle/fluid/operators/reader/reader_op_registry.h"
17
17
18
18
namespace paddle {
@@ -31,8 +31,6 @@ class DoubleBufferReader : public framework::DecoratedReader {
31
31
std::thread prefetch (
32
32
std::bind (&DoubleBufferReader::PrefetchThreadFunc, this ));
33
33
prefetch.detach ();
34
- // framework::Async(
35
- // std::bind(&DoubleBufferReader::PrefetchThreadFunc, this));
36
34
}
37
35
38
36
void ReadNext (std::vector<framework::LoDTensor>* out) override ;
You can’t perform that action at this time.
0 commit comments