Browse Source

fix(sdk/load-and-run): fix load-and-run error after adding testcases on gpu

GitOrigin-RevId: 32429d8543
tags/v1.0.0-rc1
Megvii Engine Team Xinran Xu 4 years ago
parent
commit
b91bfd59cf
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      sdk/load-and-run/dump_with_testcase_mge.py

+ 1
- 1
sdk/load-and-run/dump_with_testcase_mge.py View File

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


Loading…
Cancel
Save