提交 62753c4d 编写于 作者: M Megvii Engine Team 提交者: Xu Xinran

fix(mge/sdk): fix comp_node bug in dump_with_testcast_mge

GitOrigin-RevId: 26a8dc50b8fb63e4cbec45785cb49aa74290f8d3
上级 f1c86606
......@@ -172,6 +172,7 @@ def make_arg(
infer would be deferred to first graph execution
:param enable_static_infer: whether to enable static inference for this var
"""
comp_node = _detail.as_comp_node(comp_node)
host_val = mgb._HostSharedND(comp_node, dtype)
if value is not None:
......
......@@ -232,9 +232,8 @@ def make_feeds(args):
outputs_new = []
for i in outputs:
get = mgb.make_arg(
mge.core.graph.get_default_device(),
i.comp_node,
cg,
shape=expect_shp(i),
dtype=i.dtype,
name=expect_name(i)
)
......@@ -463,7 +462,7 @@ def main():
for testcase in feeds['testcases']:
assert isinstance(testcase, dict)
cg = mgb.comp_graph()
cn = mgb.comp_node('cpux')
cn = mgb.comp_node('xpux')
output_mgbvars = []
for name, dtype in inputs:
output_mgbvars.append(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册