diff --git a/imperative/python/megengine/jit/tracing.py b/imperative/python/megengine/jit/tracing.py index 43310e6a4de16bae42a9f18b2a639dedb2be8132..8946b88bdfe6a780c3f9c524fe80f89f8011e558 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