test compile

apache · Apr 19, 2019 · 8914f78 · 8914f78
1 parent feac9b4
commit 8914f78
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/operator/rnn.cc b/src/operator/rnn.cc
@@ -258,9 +258,11 @@ The definition of GRU here is slightly different from paper but compatible with
 .set_attr<FInferStorageType>("FInferStorageType", RNNStorageType)
 .set_attr<FCreateOpState>("FCreateOpState", CreateRNNState)
 .set_attr<FStatefulCompute>("FStatefulCompute<cpu>", RNNStatefulCompute<cpu>)
+/*
 #if MXNET_USE_MKLDNN == 1
 .set_attr<FStatefulComputeEx>("FStatefulComputeEx<cpu>", RNNStatefulComputeCPU)
 #endif
+*/
 .set_attr<nnvm::FGradient>("FGradient", RNNGrad{"_backward_RNN"})
 .set_attr<FResourceRequestEx>("FResourceRequestEx",
   [](const NodeAttrs& attrs, const int dev_mask, const DispatchMode dispatch_mode) {