"fix condition"

emailweixu-patch-1
dzhwinter 7 years ago
parent 63320f722c
commit a402d2b392

@ -132,7 +132,7 @@ class LoDTensor : public Tensor {
void set_lod(const LoD& lod) {
lod_ = lod;
if (holder_ != nullptr &&
platform::is_same_place(holder_->place(), lod.place())) {
!platform::is_same_place(holder_->place(), lod.place())) {
lod_.CopyToPeer(holder_->place());
}
}

@ -42,7 +42,13 @@ class SelectedRows {
Vector<int64_t>* mutable_rows() { return &rows_; }
void set_rows(const Vector<int64_t>& rows) { rows_ = rows; }
void set_rows(const Vector<int64_t>& rows) {
rows_ = rows;
if (value_ != nullptr &&
!platform::is_same_place(value_->place(), rows.place())) {
rows_.mutable_data(value_->place());
}
}
DDim GetCompleteDims() const {
std::vector<int64_t> dims = vectorize(value_->dims());

Loading…
Cancel
Save