saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-25 19:48:11 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Minor fix to pass x86 native build (#994)

Browse source
This commit is contained in:
jignparm 2019-05-09 00:22:45 -07:00 committed by GitHub
parent 2512b0ebeb
commit a463f765fc
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
onnxruntime/test/onnx/main.cc
View file

@ -238,14 +238,14 @@ int real_main(int argc, char* argv[], Ort::Env& env) {
#if (defined(_WIN32) && !defined(_WIN64)) || (defined(__GNUG__) && !defined(__LP64__))
//Minimize mem consumption
LoadTests(data_dirs, whitelisted_test_cases, per_sample_tolerance, relative_per_sample_tolerance, [&stat, &sf, enable_cuda, &cuda_flaky_tests](ITestCase* l) {
LoadTests(data_dirs, whitelisted_test_cases, per_sample_tolerance, relative_per_sample_tolerance, [&stat, &sf, enable_cuda, &cuda_flaky_tests, &env](ITestCase* l) {
std::unique_ptr<ITestCase> test_case_ptr(l);
if (enable_cuda && cuda_flaky_tests.find(l->GetTestCaseName()) != cuda_flaky_tests.end()) {
return;
}
TestResultStat per_case_stat;
std::vector<ITestCase*> per_case_tests = {l};
TestEnv per_case_args(per_case_tests, per_case_stat, sf);
TestEnv per_case_args(per_case_tests, per_case_stat, env ,sf);
RunTests(per_case_args, 1, 1, 1, GetDefaultThreadPool(Env::Default()));
stat += per_case_stat;
});