Browse Source

fix(mge/dump): fix dump device error with const

GitOrigin-RevId: 9dd8321fd7
release-1.5
Megvii Engine Team 3 years ago
parent
commit
d16c5caf6b
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      imperative/python/megengine/jit/tracing.py

+ 1
- 1
imperative/python/megengine/jit/tracing.py View File

@@ -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


Loading…
Cancel
Save