Unverified Commit fa38a3a6 authored by Paul Fultz II's avatar Paul Fultz II Committed by GitHub
Browse files

Merge pull request #208 from ROCmSoftwarePlatform/no_paddings

No pads in Pad op check
parents 154f21f6 319f51ae
...@@ -638,6 +638,11 @@ struct onnx_parser ...@@ -638,6 +638,11 @@ struct onnx_parser
auto&& pad_vals = attributes["pads"].ints(); auto&& pad_vals = attributes["pads"].ints();
pads = std::vector<int64_t>(pad_vals.begin(), pad_vals.end()); pads = std::vector<int64_t>(pad_vals.begin(), pad_vals.end());
} }
// check if padding is actually being done (at least one value is nonzero)
if(std::all_of(pads.begin(), pads.end(), [](const int& i) { return i == 0; }))
{
return prog.add_instruction(migraphx::op::identity{}, args.front());
}
if(contains(attributes, "value")) if(contains(attributes, "value"))
{ {
value = parse_value(attributes.at("value")).at<float>(); value = parse_value(attributes.at("value")).at<float>();
......
...@@ -683,4 +683,14 @@ TEST_CASE(logsoftmax) ...@@ -683,4 +683,14 @@ TEST_CASE(logsoftmax)
EXPECT(p == prog); EXPECT(p == prog);
} }
TEST_CASE(no_pad_test)
{
migraphx::program p;
auto l0 = p.add_parameter("0", migraphx::shape{migraphx::shape::float_type, {2, 2}});
p.add_instruction(migraphx::op::identity{}, l0);
auto prog = migraphx::parse_onnx("no_pad_test.onnx");
EXPECT(p == prog);
}
int main(int argc, const char* argv[]) { test::run(argc, argv); } int main(int argc, const char* argv[]) { test::run(argc, argv); }
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