Skip to content

Commit 386bb1f

Browse files
committed
fix crash on ds2
1 parent abce9eb commit 386bb1f

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

paddle/gserver/layers/MKLDNNLayer.h

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -339,9 +339,13 @@ class MKLDNNLayer : public Layer {
339339
* clear all grad
340340
*/
341341
void clearGrads() {
342-
output_.grad->zeroMem();
342+
if (output_.grad) {
343+
output_.grad->zeroMem();
344+
}
343345
for (size_t i = 0; i < outputOtherDevice_.size(); i++) {
344-
outputOtherDevice_[i].grad->zeroMem();
346+
if (outputOtherDevice_[i].grad) {
347+
outputOtherDevice_[i].grad->zeroMem();
348+
}
345349
}
346350
}
347351

0 commit comments

Comments
 (0)