From 65c549137fee4622268285fd4ab3cf2bf2b1718a Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Thu, 29 Oct 2020 14:09:47 +0800 Subject: [PATCH] feat(load_and_run): feat --input parameter GitOrigin-RevId: 2df8b6c164b3daffdbef20300664353832ab2ed8 --- sdk/load-and-run/src/mgblar.cpp | 15 +++++++++------ 1 file changed, 9 insertions(+), 6 deletions(-) diff --git a/sdk/load-and-run/src/mgblar.cpp b/sdk/load-and-run/src/mgblar.cpp index 96d836012..19ec721ea 100644 --- a/sdk/load-and-run/src/mgblar.cpp +++ b/sdk/load-and-run/src/mgblar.cpp @@ -807,13 +807,16 @@ void run_test_st(Args &env) { parser.feed(path); } auto inputs = parser.inputs; - for (auto& i : inputs) { - if (tensormap.find(i.first) == tensormap.end()) { - continue; - } + if (inputs.size() > 1) { + for (auto& i : inputs) { + mgb_assert(tensormap.find(i.first) != tensormap.end()); - auto& in = tensormap.find(i.first)->second; - in->copy_from(i.second); + auto& in = tensormap.find(i.first)->second; + in->copy_from(i.second); + } + } else { + auto& in = tensormap.begin()->second; + in->copy_from(inputs.begin()->second); } warmup(); -- GitLab