From 16131359903a01e9fd509ee6019ca5923983c09c Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Tue, 16 Nov 2021 17:48:19 +0800
Subject: [PATCH] fix(imperative): insert to candidate only if pinned is 0

GitOrigin-RevId: 2b849d7fe288ed22a5e5ccbadff312e104b4d2f6
---
 imperative/src/impl/interpreter/interpreter_impl.cpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/imperative/src/impl/interpreter/interpreter_impl.cpp b/imperative/src/impl/interpreter/interpreter_impl.cpp
index d80f23e7..bf122982 100644
--- a/imperative/src/impl/interpreter/interpreter_impl.cpp
+++ b/imperative/src/impl/interpreter/interpreter_impl.cpp
@@ -616,7 +616,8 @@ void ChannelImpl::produce_tensor(TensorInfo* dest, TensorPtr ptr) {
     dest->ptr = std::move(ptr);
     dest->evict_type = EvictType::NONE;
     dest->status = TensorInfo::Produced;
-    if (dest->size_exceeds_thd(state.options.dtr_evictee_minimum_size)) {
+    if (dest->pinned == 0 &&
+        dest->size_exceeds_thd(state.options.dtr_evictee_minimum_size)) {
         m_dtr.insert_candidate(dest);
     }
     notify_tensor_unsafe(dest);