Commit f027eddf authored by Fenglei Tian's avatar Fenglei Tian Committed by Robert Kimball

nbench special handle op name for "NNP_XXX" (#2883)

* remove throw from dtor (#2854)

* add special handle for the 'NNP_xx' op name

* style

* add special handle for the 'NNP_xx' op name

* style

* use description as suggest by bob
parent 5f83ff57
......@@ -78,7 +78,7 @@ multimap<size_t, string> aggregate_timing_details(const vector<PerfShape>& perf_
for (const PerfShape& p : perf_data)
{
auto node = p.get_node();
string op = node->get_name().substr(0, node->get_name().find('_'));
string op = node->description();
string shape_name = " {" + join(p.shape) + "} ";
timing[op + shape_name] += p.microseconds();
count[op + shape_name] += 1;
......@@ -98,7 +98,7 @@ multimap<size_t, string> aggregate_timing(const vector<PerfShape>& perf_data)
for (const PerfShape& p : perf_data)
{
auto node = p.get_node();
string op = node->get_name().substr(0, node->get_name().find('_'));
string op = node->description();
timing[op] += p.microseconds();
}
......@@ -355,8 +355,7 @@ OPTIONS
total_temporary_bytes += tensor->size();
total_temporary_count++;
}
string name = node->get_name();
string op_name = name.substr(0, name.find('_'));
string op_name = node->description();
string shape_name = "{" + join(node->output(0).get_shape()) + "}";
op_list[op_name + shape_name]++;
auto et = get_op_element_type(*node);
......
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