|
@@ -807,13 +807,16 @@ void run_test_st(Args &env) { |
|
|
parser.feed(path); |
|
|
parser.feed(path); |
|
|
} |
|
|
} |
|
|
auto inputs = parser.inputs; |
|
|
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(); |
|
|
warmup(); |
|
|