提交 97044062 编写于 作者: Y Yu Yang

Temporary disable async load data in PyDP2.

* It seems some other change breaks the async load. It will cause data
  error in GPU mode.
上级 8604666e
...@@ -647,7 +647,7 @@ public: ...@@ -647,7 +647,7 @@ public:
DataBatch& gpuBatch = *batch; DataBatch& gpuBatch = *batch;
std::vector<Argument>& gpuArguments = gpuBatch.getStreams(); std::vector<Argument>& gpuArguments = gpuBatch.getStreams();
gpuArguments.resize(cpuArguments.size()); gpuArguments.resize(cpuArguments.size());
gpuBatch.setSize(size); gpuBatch.setSize(bsize);
for (size_t i = 0; i < headers_.size(); ++i) { for (size_t i = 0; i < headers_.size(); ++i) {
gpuArguments[i].resizeAndCopyFrom( gpuArguments[i].resizeAndCopyFrom(
cpuArguments[i], useGpu_, HPPL_STREAM_1); cpuArguments[i], useGpu_, HPPL_STREAM_1);
......
...@@ -201,7 +201,7 @@ def define_py_data_sources2(train_list, test_list, module, obj, args=None): ...@@ -201,7 +201,7 @@ def define_py_data_sources2(train_list, test_list, module, obj, args=None):
data.load_data_module = load_data_module data.load_data_module = load_data_module
data.load_data_object = load_data_object data.load_data_object = load_data_object
data.load_data_args = load_data_args data.load_data_args = load_data_args
data.async_load_data = True data.async_load_data = False
return data return data
define_py_data_sources( define_py_data_sources(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册