Commit 470d2672 authored by Shucai Xiao's avatar Shucai Xiao
Browse files

add more test cases for softmax operator

parent 77af16d8
...@@ -592,13 +592,13 @@ struct test_softmax2 : verify_program<test_softmax2> ...@@ -592,13 +592,13 @@ struct test_softmax2 : verify_program<test_softmax2>
} }
}; };
template <int Axis> template <int Axis, migraphx::shape::type_t T>
struct test_softmax : verify_program<test_softmax<Axis>> struct test_softmax : verify_program<test_softmax<Axis, T>>
{ {
migraphx::program create_program() const migraphx::program create_program() const
{ {
migraphx::program p; migraphx::program p;
migraphx::shape s{migraphx::shape::float_type, {2080, 4, 1026, 6}}; migraphx::shape s{T, {2080, 4, 1026, 6}};
auto param = p.add_parameter("0", s); auto param = p.add_parameter("0", s);
p.add_instruction(migraphx::op::softmax{Axis}, param); p.add_instruction(migraphx::op::softmax{Axis}, param);
...@@ -606,10 +606,14 @@ struct test_softmax : verify_program<test_softmax<Axis>> ...@@ -606,10 +606,14 @@ struct test_softmax : verify_program<test_softmax<Axis>>
} }
}; };
template struct test_softmax<0>; template struct test_softmax<0, migraphx::shape::float_type>;
template struct test_softmax<1>; template struct test_softmax<1, migraphx::shape::float_type>;
template struct test_softmax<2>; template struct test_softmax<2, migraphx::shape::float_type>;
template struct test_softmax<3>; template struct test_softmax<3, migraphx::shape::float_type>;
template struct test_softmax<1, migraphx::shape::double_type>;
template struct test_softmax<3, migraphx::shape::double_type>;
template struct test_softmax<0, migraphx::shape::half_type>;
template struct test_softmax<2, migraphx::shape::half_type>;
struct test_conv : verify_program<test_conv> struct test_conv : verify_program<test_conv>
{ {
......
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