From 12c73aecb3035aeedea0de264ccfb4fb8702d309 Mon Sep 17 00:00:00 2001 From: Eric Junyuan Xie Date: Thu, 14 Jun 2018 14:39:51 -0700 Subject: [PATCH 1/2] Update cached_op.cc --- src/imperative/cached_op.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/imperative/cached_op.cc b/src/imperative/cached_op.cc index b40605bd25e2..b17fae4b3cf3 100644 --- a/src/imperative/cached_op.cc +++ b/src/imperative/cached_op.cc @@ -674,7 +674,7 @@ OpStatePtr CachedOp::StaticForward( std::lock_guard lock(state.mutex); bool match = SetForwardGraph(&state.info, recording, inputs); - match = match && state.recording != recording; + match = match && state.recording == recording; nnvm::Graph& g = state.info.fwd_graph; const auto& idx = g.indexed_graph(); From fcdca356b67261d80dbef47ea5b53e54423e742c Mon Sep 17 00:00:00 2001 From: Eric Junyuan Xie Date: Fri, 15 Jun 2018 10:59:06 -0700 Subject: [PATCH 2/2] Update cached_op.cc --- src/imperative/cached_op.cc | 1 + 1 file changed, 1 insertion(+) diff --git a/src/imperative/cached_op.cc b/src/imperative/cached_op.cc index b17fae4b3cf3..735b4cdfb770 100644 --- a/src/imperative/cached_op.cc +++ b/src/imperative/cached_op.cc @@ -676,6 +676,7 @@ OpStatePtr CachedOp::StaticForward( bool match = SetForwardGraph(&state.info, recording, inputs); match = match && state.recording == recording; + nnvm::Graph& g = state.info.fwd_graph; const auto& idx = g.indexed_graph(); if (!state.fwd_alloc || !match) {