Commit d2025171 authored by Paul's avatar Paul
Browse files

Show min/max times

parent 24d9ecbe
...@@ -190,6 +190,8 @@ struct compile_plan ...@@ -190,6 +190,8 @@ struct compile_plan
}); });
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()));
std::cout << "Fastest solution: " << config->solutions.at(i) << std::endl; std::cout << "Fastest solution: " << config->solutions.at(i) << std::endl;
std::cout << "Fastest time: " << *std::min_element(times.begin(), times.end()) << std::endl;
std::cout << "Slowest time: " << *std::max_element(times.begin(), times.end()) << std::endl;
pc.insert(preop.name(), config->problem, config->solutions.at(i)); pc.insert(preop.name(), config->problem, config->solutions.at(i));
if(not results[i].has_value()) if(not results[i].has_value())
MIGRAPHX_THROW("No valid tuned compilation."); MIGRAPHX_THROW("No valid tuned compilation.");
......
...@@ -237,7 +237,7 @@ struct find_mlir_op ...@@ -237,7 +237,7 @@ struct find_mlir_op
"recip", "recip",
"rsqrt", "rsqrt",
// There are bugs in MLIR right now for models using sigmoid so disable it for now // There are bugs in MLIR right now for models using sigmoid so disable it for now
// "sigmoid", "sigmoid",
"softmax", "softmax",
"tanh", "tanh",
}; };
......
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