From 62034fb2622b0c57a85d69eb136fed590fcbe726 Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Fri, 10 Dec 2021 19:00:19 +0800
Subject: [PATCH] fix(imperative): make CompNode finalize happens before global
 object destructor

GitOrigin-RevId: 9a1f507c693080105867486ed4574cbb3921f4fc
---
 imperative/src/impl/physical_tensor.cpp | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/imperative/src/impl/physical_tensor.cpp b/imperative/src/impl/physical_tensor.cpp
index 84035f85..f294adaf 100644
--- a/imperative/src/impl/physical_tensor.cpp
+++ b/imperative/src/impl/physical_tensor.cpp
@@ -242,6 +242,8 @@ void Tensor::static_initialize() {
     AsyncReleaser::inst();
     CompNodeSyncManager::inst();
     MultiCNConstTensorCache::inst();
+    // clean all CompNodeDepedentObjects
+    mgb_assert(!atexit(CompNode::finalize), "atexit register failed");
 }
 
 }  // namespace imperative
-- 
GitLab