From 8914f78eb7f1c351d8023c5b118e727b1776aa04 Mon Sep 17 00:00:00 2001 From: "Li, Hao H" Date: Fri, 19 Apr 2019 12:47:43 +0800 Subject: [PATCH] test compile --- src/operator/rnn.cc | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/operator/rnn.cc b/src/operator/rnn.cc index b0271ef2806a..4d66becbeba7 100644 --- 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", RNNStorageType) .set_attr("FCreateOpState", CreateRNNState) .set_attr("FStatefulCompute", RNNStatefulCompute) +/* #if MXNET_USE_MKLDNN == 1 .set_attr("FStatefulComputeEx", RNNStatefulComputeCPU) #endif +*/ .set_attr("FGradient", RNNGrad{"_backward_RNN"}) .set_attr("FResourceRequestEx", [](const NodeAttrs& attrs, const int dev_mask, const DispatchMode dispatch_mode) {