Commit cedfbd88 authored by Khalique's avatar Khalique
Browse files

formatting

parent d87962f7
...@@ -1136,9 +1136,10 @@ struct onnx_parser ...@@ -1136,9 +1136,10 @@ struct onnx_parser
{ {
std::vector<uint16_t> data_uint16(t.int32_data().begin(), t.int32_data().end()); std::vector<uint16_t> data_uint16(t.int32_data().begin(), t.int32_data().end());
std::vector<half> data_half; std::vector<half> data_half;
std::transform(data_uint16.begin(), data_uint16.end(), std::back_inserter(data_half), [&](uint16_t raw_val){ std::transform(data_uint16.begin(),
return *reinterpret_cast<half*>(&raw_val); data_uint16.end(),
}); std::back_inserter(data_half),
[&](uint16_t raw_val) { return *reinterpret_cast<half*>(&raw_val); });
return literal{{shape::half_type, dims}, data_half.begin(), data_half.end()}; return literal{{shape::half_type, dims}, data_half.begin(), data_half.end()};
} }
case onnx::TensorProto::DOUBLE: case onnx::TensorProto::DOUBLE:
......
...@@ -1218,9 +1218,11 @@ TEST_CASE(lrn_test) ...@@ -1218,9 +1218,11 @@ TEST_CASE(lrn_test)
TEST_CASE(add_fp16_test) TEST_CASE(add_fp16_test)
{ {
migraphx::program p ; migraphx::program p;
auto l0 = p.add_literal(migraphx::literal{migraphx::shape{migraphx::shape::half_type, {1}}, {1.5}}); auto l0 =
auto l1 = p.add_literal(migraphx::literal{migraphx::shape{migraphx::shape::half_type, {1}}, {2.5}}); p.add_literal(migraphx::literal{migraphx::shape{migraphx::shape::half_type, {1}}, {1.5}});
auto l1 =
p.add_literal(migraphx::literal{migraphx::shape{migraphx::shape::half_type, {1}}, {2.5}});
p.add_instruction(migraphx::op::add{}, l0, l1); p.add_instruction(migraphx::op::add{}, l0, l1);
auto prog = migraphx::parse_onnx("add_fp16_test.onnx"); auto prog = migraphx::parse_onnx("add_fp16_test.onnx");
......
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