fix.

apache · Sep 12, 2018 · 7e722a9 · 7e722a9
1 parent ded2cc2
commit 7e722a9
Showing 1 changed file with 2 additions and 8 deletions.
diff --git a/src/imperative/cached_op.cc b/src/imperative/cached_op.cc
@@ -1339,10 +1339,7 @@ NNVM_REGISTER_OP(_CachedOp)
     const CachedOpPtr& op = nnvm::get<CachedOpPtr>(attrs.parsed);
     return op::DefaultSubgraphOpResourceRequest1(op->GetForwardSym());
   })
-.set_attr<FExecType>("FExecType",
-  [](const nnvm::NodeAttrs& attrs) {
-    return ExecType::kSubgraphExec;
-  })
+.set_attr<FExecType>("FExecType", DefaultSubgraphOpExecType)
 .add_argument("data", "NDArray-or-Symbol[]", "input data list");
 
 NNVM_REGISTER_OP(_backward_CachedOp)
@@ -1364,10 +1361,7 @@ NNVM_REGISTER_OP(_backward_CachedOp)
   })
 .set_attr<FStatefulComputeEx>("FStatefulComputeEx<cpu>", CachedOpBackward)
 .set_attr<FStatefulComputeEx>("FStatefulComputeEx<gpu>", CachedOpBackward)
-.set_attr<FExecType>("FExecType",
-  [](const nnvm::NodeAttrs& attrs) {
-    return ExecType::kSubgraphExec;
-  })
+.set_attr<FExecType>("FExecType", DefaultSubgraphOpExecType)
 .set_attr<bool>("TIsLayerOpBackward", true)
 .set_attr<bool>("TIsBackward", true);