Commit 82b7ccfb authored by Chris Austen's avatar Chris Austen
Browse files

fix file formating issue

parent 64998903
...@@ -211,21 +211,21 @@ struct find_mlir_op ...@@ -211,21 +211,21 @@ struct find_mlir_op
} }
const std::initializer_list<std::string> any_type_ops = {"@literal", "@param", "@return"}; const std::initializer_list<std::string> any_type_ops = {"@literal", "@param", "@return"};
const std::initializer_list<std::string> no_bool_ops = {"convolution", const std::initializer_list<std::string> no_bool_ops = {"convolution",
"quant_convolution", "quant_convolution",
"dot", "dot",
"quant_dot", "quant_dot",
"add", "add",
"clip", "clip",
"relu", "relu",
"sub", "sub",
"mul", "mul",
"div", "div",
"pow", "pow",
"where", "where",
"quantizelinear", "quantizelinear",
"dequantizelinear", "dequantizelinear",
"abs", "abs",
"neg"}; "neg"};
const std::initializer_list<std::string> fp_only_ops = {"ceil", const std::initializer_list<std::string> fp_only_ops = {"ceil",
"erf", "erf",
"exp", "exp",
......
...@@ -218,7 +218,7 @@ TEST_CASE(single_target_multi_compile) ...@@ -218,7 +218,7 @@ TEST_CASE(single_target_multi_compile)
auto score_threshold = gpu_mod->add_literal(0.0f); auto score_threshold = gpu_mod->add_literal(0.0f);
auto r = gpu_mod->add_instruction( auto r = gpu_mod->add_instruction(
migraphx::make_op("nonmaxsuppression", migraphx::make_op("nonmaxsuppression",
{{"center_point_box", true}, {"use_dyn_output", true}}), {{"center_point_box", true}, {"use_dyn_output", true}}),
boxes_param_gpu, boxes_param_gpu,
scores_l, scores_l,
max_out_l, max_out_l,
...@@ -241,7 +241,7 @@ TEST_CASE(single_target_multi_compile) ...@@ -241,7 +241,7 @@ TEST_CASE(single_target_multi_compile)
// eval // eval
migraphx::parameter_map params; migraphx::parameter_map params;
std::vector<float> boxes_vec = {0.5, 0.5, 1.0, 1.0, 0.5, 0.6, 1.0, 1.0, 0.5, 0.4, 1.0, 1.0, std::vector<float> boxes_vec = {0.5, 0.5, 1.0, 1.0, 0.5, 0.6, 1.0, 1.0, 0.5, 0.4, 1.0, 1.0,
0.5, 10.5, 1.0, 1.0, 0.5, 10.6, 1.0, 1.0, 0.5, 100.5, 1.0, 1.0}; 0.5, 10.5, 1.0, 1.0, 0.5, 10.6, 1.0, 1.0, 0.5, 100.5, 1.0, 1.0};
params["boxes"] = migraphx::argument(boxes_s, boxes_vec.data()); params["boxes"] = migraphx::argument(boxes_s, boxes_vec.data());
auto output = p.eval(params).back(); auto output = p.eval(params).back();
std::vector<int64_t> gold_vec = {0, 0, 3, 0, 0, 0, 0, 0, 5}; std::vector<int64_t> gold_vec = {0, 0, 3, 0, 0, 0, 0, 0, 5};
......
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