From d16c5caf6bf43731d1f6bdefd951282a8495b25b Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Fri, 23 Jul 2021 19:57:05 +0800
Subject: [PATCH] fix(mge/dump): fix dump device error with const

GitOrigin-RevId: 9dd8321fd7f3ab09d8b7e3224ffa2740625bbb32
---
 imperative/python/megengine/jit/tracing.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/imperative/python/megengine/jit/tracing.py b/imperative/python/megengine/jit/tracing.py
index 43310e6a..8946b88b 100644
--- a/imperative/python/megengine/jit/tracing.py
+++ b/imperative/python/megengine/jit/tracing.py
@@ -837,7 +837,7 @@ class trace:
                     h2v[h] = graph.make_const(
                         info.bound_data.numpy(),
                         dtype=info.dtype,
-                        device=info.device,
+                        device=dumped_device(info),
                         name=info.name,
                     )
                 continue