"test/gpu/ops_test.cpp" did not exist on "51724da56684c53146af2029f16846f5bd956129"
Commit 8c3a42e7 authored by Shucai Xiao's avatar Shucai Xiao
Browse files

Add onnx test example for operators asin acos atan sinh and cosh.

parent d463391e
......@@ -54,6 +54,11 @@ struct onnx_parser
// disable dropout for inference
add_generic_op("Dropout", op::identity{});
add_generic_op("Identity", op::identity{});
add_generic_op("Sinh", op::sinh{});
add_generic_op("Cosh", op::cosh{});
add_generic_op("Asin", op::asin{});
add_generic_op("Acos", op::acos{});
add_generic_op("Atan", op::atan{});
add_broadcastable_binary_op("Add", op::add{});
add_broadcastable_binary_op("Div", op::div{});
......
......@@ -10,4 +10,4 @@ b


B
\ No newline at end of file
B
\ No newline at end of file
 cosh-example:;
xy"Cosh test_coshZ
x

b
y

B
\ No newline at end of file
......@@ -169,6 +169,61 @@ void dropout_test()
EXPECT(p == prog);
}
void sinh_test()
{
migraphx::program p;
auto input = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {10}});
p.add_instruction(migraphx::op::sinh{}, input);
auto prog = migraphx::parse_onnx("sinh_test.onnx");
EXPECT(p == prog);
}
void cosh_test()
{
migraphx::program p;
auto input = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {1}});
p.add_instruction(migraphx::op::cosh{}, input);
auto prog = migraphx::parse_onnx("cosh_test.onnx");
EXPECT(p == prog);
}
void asin_test()
{
migraphx::program p;
auto input = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {10}});
p.add_instruction(migraphx::op::asin{}, input);
auto prog = migraphx::parse_onnx("asin_test.onnx");
EXPECT(p == prog);
}
void acos_test()
{
migraphx::program p;
auto input = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {10}});
p.add_instruction(migraphx::op::acos{}, input);
auto prog = migraphx::parse_onnx("acos_test.onnx");
EXPECT(p == prog);
}
void atan_test()
{
migraphx::program p;
auto input = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {10}});
p.add_instruction(migraphx::op::atan{}, input);
auto prog = migraphx::parse_onnx("atan_test.onnx");
EXPECT(p == prog);
}
int main()
{
pytorch_conv_bias_test();
......@@ -181,4 +236,9 @@ int main()
globalmaxpool_test();
transpose_test();
dropout_test();
sinh_test();
cosh_test();
asin_test();
acos_test();
atan_test();
}
 sinh-example:;
xy"Sinh test_sinhZ
x


b
y


B
\ No newline at end of file
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