From 38a5c1c9706238b5cb37679ad0a159d780bb5f64 Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Sun, 6 Sep 2020 15:23:14 +0800
Subject: [PATCH] fix(mge): fix refcnt in TensorShape caster

GitOrigin-RevId: 24f6c7a6d0401c24496fc68e6d05147d1a50ca2a
---
 imperative/python/src/helper.h | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/imperative/python/src/helper.h b/imperative/python/src/helper.h
index 82cd083f..ec0942c1 100644
--- a/imperative/python/src/helper.h
+++ b/imperative/python/src/helper.h
@@ -279,7 +279,7 @@ namespace detail {
         PYBIND11_TYPE_CASTER(mgb::TensorShape, _("TensorShape"));
     public:
         bool load(handle src, bool convert) {
-            auto obj = reinterpret_steal<object>(src);
+            auto obj = reinterpret_borrow<object>(src);
             if (!convert && !isinstance<tuple>(obj)) {
                 return false;
             }