diff --git a/src/operator/nn/dropout-inl.h b/src/operator/nn/dropout-inl.h index 2a643a266b2b..cb02be6e0338 100644 --- a/src/operator/nn/dropout-inl.h +++ b/src/operator/nn/dropout-inl.h @@ -398,6 +398,8 @@ class DropoutOp { } } } else { + if (req[dropout::kOut] == kWriteInplace) return; + MXNET_ASSIGN_REQ_SWITCH(req[dropout::kOut], Req, { mxnet_op::Kernel, xpu>::Launch( s, out.Size(), out.dptr(), in.dptr());