From b91bfd59cfacfc7ee2f9aa14ad42cf34b791d783 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Mon, 24 Aug 2020 18:47:04 +0800 Subject: [PATCH] fix(sdk/load-and-run): fix load-and-run error after adding testcases on gpu GitOrigin-RevId: 32429d85432e12aad8be0dfd8b491afb78b707ad --- sdk/load-and-run/dump_with_testcase_mge.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sdk/load-and-run/dump_with_testcase_mge.py b/sdk/load-and-run/dump_with_testcase_mge.py index 8180a7f9..bef22e0f 100755 --- a/sdk/load-and-run/dump_with_testcase_mge.py +++ b/sdk/load-and-run/dump_with_testcase_mge.py @@ -462,7 +462,7 @@ def main(): for testcase in feeds['testcases']: assert isinstance(testcase, dict) cg = mgb.comp_graph() - cn = mgb.comp_node('xpux') + cn = mgb.comp_node('cpux') output_mgbvars = [] for name, dtype in inputs: output_mgbvars.append(