fix bug

apache · May 16, 2019 · b5a7f1e · b5a7f1e
1 parent dd09228
commit b5a7f1e
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/src/operator/nn/layer_norm.cu b/src/operator/nn/layer_norm.cu
@@ -667,7 +667,6 @@ void LayerNormGradCompute<gpu>(const nnvm::NodeAttrs& attrs,
                                const std::vector<OpReqType>& req,
                                const std::vector<TBlob>& outputs) {
   const LayerNormParam& param = nnvm::get<LayerNormParam>(attrs.parsed);
-  if (req[0] == kNullOp) return;
   int axis = param.axis;
   if (axis < 0) {
     axis += static_cast<int>(inputs[0].ndim());