Merge pull request #5010 from tensor-tang/merge

fix crash on ds2
revert-4814-Add_sequence_project_op
Tao Luo 7 years ago committed by GitHub
commit 48173e8599

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

Loading…
Cancel
Save