fix crash on ds2

revert-4814-Add_sequence_project_op
tensor-tang 7 years ago
parent abce9eb750
commit 386bb1fece

@ -339,9 +339,13 @@ private:
* clear all grad * clear all grad
*/ */
void clearGrads() { void clearGrads() {
output_.grad->zeroMem(); if (output_.grad) {
output_.grad->zeroMem();
}
for (size_t i = 0; i < outputOtherDevice_.size(); i++) { for (size_t i = 0; i < outputOtherDevice_.size(); i++) {
outputOtherDevice_[i].grad->zeroMem(); if (outputOtherDevice_[i].grad) {
outputOtherDevice_[i].grad->zeroMem();
}
} }
} }

Loading…
Cancel
Save