提交 d16c5caf 编写于 作者: M Megvii Engine Team

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

GitOrigin-RevId: 9dd8321fd7f3ab09d8b7e3224ffa2740625bbb32
上级 2c4ff543
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册