Skip to content

P
Paddle

PaddlePaddle / Paddle
1 年多 前同步成功

通知 2302
Star 20931
Fork 5422
P
Paddle
未验证 提交 0d2ccfbd 编写于 作者: Y yuyang18
浏览文件
操作

Remove atomic

上级 b4f0e579
显示空白变更内容
内联 并排
Showing with 2 addition and 5 deletion
paddle/fluid/framework/reader.cc
浏览文件 @ 0d2ccfbd
...@@ -18,9 +18,7 @@ namespace paddle { ...@@ -18,9 +18,7 @@ namespace paddle {
namespace framework { namespace framework {
void ReaderBase::ReadNext(std::vector<LoDTensor> *out) { void ReaderBase::ReadNext(std::vector<LoDTensor> *out) {
if (status_ != ReaderStatus::kRunning) { PADDLE_ENFORCE_EQ(status_, ReaderStatus::kRunning);
PADDLE_THROW("The reader is not at the status of 'running'.");
}
ReadNextImpl(out); ReadNextImpl(out);
} }
......
paddle/fluid/framework/reader.h
浏览文件 @ 0d2ccfbd
...@@ -14,7 +14,6 @@ ...@@ -14,7 +14,6 @@
#pragma once #pragma once
#include <atomic>
#include <memory> #include <memory>
#include <vector> #include <vector>
...@@ -44,7 +43,7 @@ class ReaderBase { ...@@ -44,7 +43,7 @@ class ReaderBase {
virtual void StartImpl() = 0; virtual void StartImpl() = 0;
std::atomic<ReaderStatus> status_{kRunning}; ReaderStatus status_{kRunning};
}; };
class DecoratedReader : public ReaderBase { class DecoratedReader : public ReaderBase {
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或