From 886e7c6eb6b9cbec621940d14298f52f8e80f71b Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Thu, 4 Feb 2021 13:52:52 +0800
Subject: [PATCH] refactor(mge/tensor): support scalar in detach and grad

GitOrigin-RevId: 38349c8357871d853219026d1431b98ca55fe3a4
---
 imperative/python/src/tensor.cpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/imperative/python/src/tensor.cpp b/imperative/python/src/tensor.cpp
index 44c4099e..0064192c 100644
--- a/imperative/python/src/tensor.cpp
+++ b/imperative/python/src/tensor.cpp
@@ -495,9 +495,10 @@ PyObject* TensorWrapper::detach() {
         new_tensor = std::make_shared<Tensor>(m_tensor->m_var);
     }
     new_tensor->m_trace_info = m_tensor->m_trace_info;
+
+    new_tensor->m_flags = m_tensor->m_flags;
     auto ret = TensorWrapper::make(pytype, std::move(new_tensor));
     return ret.release().ptr();
-
 }
 
 PyObject* TensorWrapper::_dev_tensor(){