fallback mkldnn fc bwd in imperative mode (#16672)

apache · Oct 30, 2019 · e663ce1 · e663ce1
1 parent b5d4f87
commit e663ce1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/operator/nn/fully_connected.cc b/src/operator/nn/fully_connected.cc
@@ -147,7 +147,8 @@ void FullyConnectedGradComputeExCPU(const nnvm::NodeAttrs& attrs,
                                     const std::vector<NDArray> &inputs,
                                     const std::vector<OpReqType> &req,
                                     const std::vector<NDArray> &outputs) {
-  if (SupportMKLDNNFC(inputs[0])) {
+  bool mkldnn_fc_backward_enable = false;
+  if (SupportMKLDNNFC(inputs[0]) && mkldnn_fc_backward_enable) {
     MKLDNN_OPCHECK_INIT(true, outputs.size(), inputs, outputs);
     MKLDNNFCBackward(attrs, ctx, inputs, req, outputs);
     MKLDNN_OPCHECK_RUN(FullyConnectedGradCompute<cpu>, attrs, ctx, inputs, req,