fix an error

shanyi15-patch-2
fengjiayi 7 years ago
parent 4fb7b96756
commit e8d21b6349

@ -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)

@ -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 <thread>
#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<framework::LoDTensor>* out) override;

Loading…
Cancel
Save