Commit 9b856dd8 authored by Paul's avatar Paul
Browse files

Format

parent 149e783c
...@@ -180,9 +180,10 @@ struct compile_plan ...@@ -180,9 +180,10 @@ struct compile_plan
times.reserve(results.size()); times.reserve(results.size());
std::transform( std::transform(
results.begin(), results.end(), std::back_inserter(times), [&](const auto& cr) { results.begin(), results.end(), std::back_inserter(times), [&](const auto& cr) {
if (not cr.has_value()) if(not cr.has_value())
return std::numeric_limits<double>::max(); return std::numeric_limits<double>::max();
return time_op(*ctx, cr->replace.code_object, to_shapes(cr->ins->inputs()), 20).first; return time_op(*ctx, cr->replace.code_object, to_shapes(cr->ins->inputs()), 20)
.first;
}); });
auto i = std::distance(times.begin(), std::min_element(times.begin(), times.end())); auto i = std::distance(times.begin(), std::min_element(times.begin(), times.end()));
pc.insert(preop.name(), config->problem, config->solutions.at(i)); pc.insert(preop.name(), config->problem, config->solutions.at(i));
...@@ -203,7 +204,7 @@ void par_compile(std::size_t n, F f) ...@@ -203,7 +204,7 @@ void par_compile(std::size_t n, F f)
if(n == 0) if(n == 0)
return; return;
auto d = value_of(MIGRAPHX_GPU_COMPILE_PARALLEL{}); auto d = value_of(MIGRAPHX_GPU_COMPILE_PARALLEL{});
if (d == 0) if(d == 0)
d = n; d = n;
par_for(n, n / d, f); par_for(n, n / d, f);
} }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment