未验证 提交 4416c793 编写于 作者: W wangguanqun 提交者: GitHub

fix set_lod in data_feed (#35000)

* add trainer desc config to distributed strategy

* code style modified

* data_feed set lod
上级 4c115a82
...@@ -847,8 +847,10 @@ void MultiSlotDataFeed::PutToFeedVec( ...@@ -847,8 +847,10 @@ void MultiSlotDataFeed::PutToFeedVec(
total_instance * sizeof(int64_t)); total_instance * sizeof(int64_t));
} }
if (!use_slots_is_dense_[i]) {
LoD data_lod{offset}; LoD data_lod{offset};
feed_vec_[i]->set_lod(data_lod); feed_vec_[i]->set_lod(data_lod);
}
if (use_slots_is_dense_[i]) { if (use_slots_is_dense_[i]) {
if (inductive_shape_index_[i] != -1) { if (inductive_shape_index_[i] != -1) {
use_slots_shape_[i][inductive_shape_index_[i]] = use_slots_shape_[i][inductive_shape_index_[i]] =
...@@ -1206,8 +1208,10 @@ void MultiSlotInMemoryDataFeed::PutToFeedVec( ...@@ -1206,8 +1208,10 @@ void MultiSlotInMemoryDataFeed::PutToFeedVec(
} }
auto& slot_offset = offset_[i]; auto& slot_offset = offset_[i];
if (this->input_type_ == 0) { if (this->input_type_ == 0) {
if (!use_slots_is_dense_[i]) {
LoD data_lod{slot_offset}; LoD data_lod{slot_offset};
feed_vec_[i]->set_lod(data_lod); feed_vec_[i]->set_lod(data_lod);
}
} else if (this->input_type_ == 1) { } else if (this->input_type_ == 1) {
if (!use_slots_is_dense_[i]) { if (!use_slots_is_dense_[i]) {
std::vector<size_t> tmp_offset; std::vector<size_t> tmp_offset;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册