Avoid memory copy for dropout inference (#15521)

* fix dropout for inference
apache · Jul 15, 2019 · 9724ce6 · 9724ce6
1 parent b88705e
commit 9724ce6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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<mxnet_op::op_with_req<mshadow_op::identity, Req>, xpu>::Launch(
             s, out.Size(), out.dptr<DType>(), in.dptr<DType>());