Commit 94e3a2e4 authored by Shucai Xiao's avatar Shucai Xiao
Browse files

change size_t to int

parent 26bd92d8
...@@ -8,8 +8,8 @@ struct test_conv_bn_add : verify_program<test_conv_bn_add> ...@@ -8,8 +8,8 @@ struct test_conv_bn_add : verify_program<test_conv_bn_add>
{ {
static migraphx::instruction_ref add_bn(migraphx::module& m, static migraphx::instruction_ref add_bn(migraphx::module& m,
migraphx::instruction_ref x, migraphx::instruction_ref x,
std::size_t channels, int channels,
std::size_t seed = 1) int seed = 1)
{ {
migraphx::shape vars{migraphx::shape::float_type, {channels}}; migraphx::shape vars{migraphx::shape::float_type, {channels}};
auto scale = m.add_literal(migraphx::abs(migraphx::generate_literal(vars, 1 + seed))); auto scale = m.add_literal(migraphx::abs(migraphx::generate_literal(vars, 1 + seed)));
...@@ -24,8 +24,8 @@ struct test_conv_bn_add : verify_program<test_conv_bn_add> ...@@ -24,8 +24,8 @@ struct test_conv_bn_add : verify_program<test_conv_bn_add>
{ {
migraphx::program p; migraphx::program p;
auto* mm = p.get_main_module(); auto* mm = p.get_main_module();
std::size_t ichannels = 64; int ichannels = 64;
std::size_t ochannels = 256; int ochannels = 256;
auto x = mm->add_parameter("x", {migraphx::shape::float_type, {1, ichannels, 56, 56}}); auto x = mm->add_parameter("x", {migraphx::shape::float_type, {1, ichannels, 56, 56}});
auto w = mm->add_literal(migraphx::generate_literal( auto w = mm->add_literal(migraphx::generate_literal(
{migraphx::shape::float_type, {ochannels, ichannels, 1, 1}}, 1)); {migraphx::shape::float_type, {ochannels, ichannels, 1, 1}}, 1));
......
...@@ -7,7 +7,7 @@ ...@@ -7,7 +7,7 @@
struct test_conv_bn_relu_pooling2 : verify_program<test_conv_bn_relu_pooling2> struct test_conv_bn_relu_pooling2 : verify_program<test_conv_bn_relu_pooling2>
{ {
static migraphx::instruction_ref static migraphx::instruction_ref
add_bn(migraphx::program& p, migraphx::instruction_ref x, std::size_t channels) add_bn(migraphx::program& p, migraphx::instruction_ref x, int channels)
{ {
auto* mm = p.get_main_module(); auto* mm = p.get_main_module();
migraphx::shape vars{migraphx::shape::float_type, {channels}}; migraphx::shape vars{migraphx::shape::float_type, {channels}};
......
...@@ -10,7 +10,7 @@ struct test_gelu : verify_program<test_gelu> ...@@ -10,7 +10,7 @@ struct test_gelu : verify_program<test_gelu>
{ {
migraphx::program p; migraphx::program p;
auto* mm = p.get_main_module(); auto* mm = p.get_main_module();
std::vector<size_t> input_lens{1, 1, 5}; std::vector<int> input_lens{1, 1, 5};
auto x = mm->add_parameter("x", {migraphx::shape::float_type, input_lens}); auto x = mm->add_parameter("x", {migraphx::shape::float_type, input_lens});
auto half = mm->add_literal(0.5f); auto half = mm->add_literal(0.5f);
auto one = mm->add_literal(1.0f); auto one = mm->add_literal(1.0f);
......
...@@ -12,11 +12,11 @@ struct test_gru_bidirct : verify_program<test_gru_bidirct> ...@@ -12,11 +12,11 @@ struct test_gru_bidirct : verify_program<test_gru_bidirct>
{ {
migraphx::program create_program() const migraphx::program create_program() const
{ {
std::size_t batch_size = 2; int batch_size = 2;
std::size_t seq_len = 3; int seq_len = 3;
std::size_t hidden_size = 5; int hidden_size = 5;
std::size_t input_size = 8; int input_size = 8;
std::size_t num_dirct = 2; int num_dirct = 2;
float clip = 0.0f; float clip = 0.0f;
migraphx::program p; migraphx::program p;
......
...@@ -12,11 +12,11 @@ struct test_gru_bidirct_3args : verify_program<test_gru_bidirct_3args> ...@@ -12,11 +12,11 @@ struct test_gru_bidirct_3args : verify_program<test_gru_bidirct_3args>
{ {
migraphx::program create_program() const migraphx::program create_program() const
{ {
std::size_t batch_size = 2; int batch_size = 2;
std::size_t seq_len = 3; int seq_len = 3;
std::size_t hidden_size = 5; int hidden_size = 5;
std::size_t input_size = 8; int input_size = 8;
std::size_t num_dirct = 2; int num_dirct = 2;
float clip = 0.0f; float clip = 0.0f;
migraphx::program p; migraphx::program p;
......
...@@ -12,11 +12,11 @@ struct test_gru_bidirct_3args_und : verify_program<test_gru_bidirct_3args_und> ...@@ -12,11 +12,11 @@ struct test_gru_bidirct_3args_und : verify_program<test_gru_bidirct_3args_und>
{ {
migraphx::program create_program() const migraphx::program create_program() const
{ {
std::size_t batch_size = 2; int batch_size = 2;
std::size_t seq_len = 3; int seq_len = 3;
std::size_t hidden_size = 5; int hidden_size = 5;
std::size_t input_size = 8; int input_size = 8;
std::size_t num_dirct = 2; int num_dirct = 2;
float clip = 0.0f; float clip = 0.0f;
migraphx::program p; migraphx::program p;
......
...@@ -12,11 +12,11 @@ struct test_gru_bidirct_default_actv : verify_program<test_gru_bidirct_default_a ...@@ -12,11 +12,11 @@ struct test_gru_bidirct_default_actv : verify_program<test_gru_bidirct_default_a
{ {
migraphx::program create_program() const migraphx::program create_program() const
{ {
std::size_t batch_size = 2; int batch_size = 2;
std::size_t seq_len = 1; int seq_len = 1;
std::size_t hidden_size = 5; int hidden_size = 5;
std::size_t input_size = 8; int input_size = 8;
std::size_t num_dirct = 2; int num_dirct = 2;
float clip = 0.0f; float clip = 0.0f;
migraphx::program p; migraphx::program p;
......
...@@ -12,11 +12,11 @@ struct test_gru_bidirct_default_actv1 : verify_program<test_gru_bidirct_default_ ...@@ -12,11 +12,11 @@ struct test_gru_bidirct_default_actv1 : verify_program<test_gru_bidirct_default_
{ {
migraphx::program create_program() const migraphx::program create_program() const
{ {
std::size_t batch_size = 2; int batch_size = 2;
std::size_t seq_len = 3; int seq_len = 3;
std::size_t hidden_size = 5; int hidden_size = 5;
std::size_t input_size = 8; int input_size = 8;
std::size_t num_dirct = 2; int num_dirct = 2;
float clip = 0.0f; float clip = 0.0f;
migraphx::program p; migraphx::program p;
......
...@@ -12,11 +12,11 @@ struct test_gru_bidirct_seq1 : verify_program<test_gru_bidirct_seq1> ...@@ -12,11 +12,11 @@ struct test_gru_bidirct_seq1 : verify_program<test_gru_bidirct_seq1>
{ {
migraphx::program create_program() const migraphx::program create_program() const
{ {
std::size_t batch_size = 2; int batch_size = 2;
std::size_t seq_len = 1; int seq_len = 1;
std::size_t hidden_size = 5; int hidden_size = 5;
std::size_t input_size = 8; int input_size = 8;
std::size_t num_dirct = 2; int num_dirct = 2;
float clip = 0.0f; float clip = 0.0f;
migraphx::program p; migraphx::program p;
......
...@@ -12,11 +12,11 @@ struct test_gru_forward : verify_program<test_gru_forward> ...@@ -12,11 +12,11 @@ struct test_gru_forward : verify_program<test_gru_forward>
{ {
migraphx::program create_program() const migraphx::program create_program() const
{ {
std::size_t batch_size = 2; int batch_size = 2;
std::size_t seq_len = 3; int seq_len = 3;
std::size_t hidden_size = 5; int hidden_size = 5;
std::size_t input_size = 8; int input_size = 8;
std::size_t num_dirct = 1; int num_dirct = 1;
float clip = 0.0f; float clip = 0.0f;
migraphx::program p; migraphx::program p;
......
...@@ -12,11 +12,11 @@ struct test_gru_forward_3args : verify_program<test_gru_forward_3args> ...@@ -12,11 +12,11 @@ struct test_gru_forward_3args : verify_program<test_gru_forward_3args>
{ {
migraphx::program create_program() const migraphx::program create_program() const
{ {
std::size_t batch_size = 2; int batch_size = 2;
std::size_t seq_len = 3; int seq_len = 3;
std::size_t hidden_size = 5; int hidden_size = 5;
std::size_t input_size = 8; int input_size = 8;
std::size_t num_dirct = 1; int num_dirct = 1;
float clip = 0.0f; float clip = 0.0f;
migraphx::program p; migraphx::program p;
......
...@@ -12,11 +12,11 @@ struct test_gru_forward_3args_und : verify_program<test_gru_forward_3args_und> ...@@ -12,11 +12,11 @@ struct test_gru_forward_3args_und : verify_program<test_gru_forward_3args_und>
{ {
migraphx::program create_program() const migraphx::program create_program() const
{ {
std::size_t batch_size = 2; int batch_size = 2;
std::size_t seq_len = 3; int seq_len = 3;
std::size_t hidden_size = 5; int hidden_size = 5;
std::size_t input_size = 8; int input_size = 8;
std::size_t num_dirct = 1; int num_dirct = 1;
float clip = 0.0f; float clip = 0.0f;
migraphx::program p; migraphx::program p;
......
...@@ -12,11 +12,11 @@ struct test_gru_forward_default_actv : verify_program<test_gru_forward_default_a ...@@ -12,11 +12,11 @@ struct test_gru_forward_default_actv : verify_program<test_gru_forward_default_a
{ {
migraphx::program create_program() const migraphx::program create_program() const
{ {
std::size_t batch_size = 2; int batch_size = 2;
std::size_t seq_len = 1; int seq_len = 1;
std::size_t hidden_size = 5; int hidden_size = 5;
std::size_t input_size = 8; int input_size = 8;
std::size_t num_dirct = 1; int num_dirct = 1;
float clip = 0.0f; float clip = 0.0f;
migraphx::program p; migraphx::program p;
......
...@@ -12,11 +12,11 @@ struct test_gru_forward_default_actv1 : verify_program<test_gru_forward_default_ ...@@ -12,11 +12,11 @@ struct test_gru_forward_default_actv1 : verify_program<test_gru_forward_default_
{ {
migraphx::program create_program() const migraphx::program create_program() const
{ {
std::size_t batch_size = 2; int batch_size = 2;
std::size_t seq_len = 3; int seq_len = 3;
std::size_t hidden_size = 5; int hidden_size = 5;
std::size_t input_size = 8; int input_size = 8;
std::size_t num_dirct = 1; int num_dirct = 1;
float clip = 0.0f; float clip = 0.0f;
migraphx::program p; migraphx::program p;
......
...@@ -12,11 +12,11 @@ struct test_gru_forward_seq1 : verify_program<test_gru_forward_seq1> ...@@ -12,11 +12,11 @@ struct test_gru_forward_seq1 : verify_program<test_gru_forward_seq1>
{ {
migraphx::program create_program() const migraphx::program create_program() const
{ {
std::size_t batch_size = 2; int batch_size = 2;
std::size_t seq_len = 1; int seq_len = 1;
std::size_t hidden_size = 5; int hidden_size = 5;
std::size_t input_size = 8; int input_size = 8;
std::size_t num_dirct = 1; int num_dirct = 1;
float clip = 0.0f; float clip = 0.0f;
migraphx::program p; migraphx::program p;
......
...@@ -12,11 +12,11 @@ struct test_gru_reverse_3args : verify_program<test_gru_reverse_3args> ...@@ -12,11 +12,11 @@ struct test_gru_reverse_3args : verify_program<test_gru_reverse_3args>
{ {
migraphx::program create_program() const migraphx::program create_program() const
{ {
std::size_t batch_size = 2; int batch_size = 2;
std::size_t seq_len = 3; int seq_len = 3;
std::size_t hidden_size = 5; int hidden_size = 5;
std::size_t input_size = 8; int input_size = 8;
std::size_t num_dirct = 1; int num_dirct = 1;
float clip = 0.0f; float clip = 0.0f;
migraphx::program p; migraphx::program p;
......
...@@ -12,11 +12,11 @@ struct test_gru_reverse_last : verify_program<test_gru_reverse_last> ...@@ -12,11 +12,11 @@ struct test_gru_reverse_last : verify_program<test_gru_reverse_last>
{ {
migraphx::program create_program() const migraphx::program create_program() const
{ {
std::size_t batch_size = 2; int batch_size = 2;
std::size_t seq_len = 3; int seq_len = 3;
std::size_t hidden_size = 5; int hidden_size = 5;
std::size_t input_size = 8; int input_size = 8;
std::size_t num_dirct = 1; int num_dirct = 1;
float clip = 0.0f; float clip = 0.0f;
migraphx::program p; migraphx::program p;
......
...@@ -12,11 +12,11 @@ struct test_gru_two_outputs : verify_program<test_gru_two_outputs> ...@@ -12,11 +12,11 @@ struct test_gru_two_outputs : verify_program<test_gru_two_outputs>
{ {
migraphx::program create_program() const migraphx::program create_program() const
{ {
std::size_t batch_size = 2; int batch_size = 2;
std::size_t seq_len = 1; int seq_len = 1;
std::size_t hidden_size = 5; int hidden_size = 5;
std::size_t input_size = 8; int input_size = 8;
std::size_t num_dirct = 1; int num_dirct = 1;
float clip = 0.0f; float clip = 0.0f;
migraphx::program p; migraphx::program p;
......
...@@ -7,7 +7,7 @@ ...@@ -7,7 +7,7 @@
#include <migraphx/op/reduce_mean.hpp> #include <migraphx/op/reduce_mean.hpp>
migraphx::instruction_ref migraphx::instruction_ref
add_layernorm(migraphx::module& m, migraphx::instruction_ref x, std::vector<size_t> dims) add_layernorm(migraphx::module& m, migraphx::instruction_ref x, std::vector<int> dims)
{ {
auto scale = auto scale =
m.add_parameter("scale", migraphx::shape{migraphx::shape::float_type, {dims.back()}}); m.add_parameter("scale", migraphx::shape{migraphx::shape::float_type, {dims.back()}});
...@@ -45,7 +45,7 @@ struct test_layernorm : verify_program<test_layernorm> ...@@ -45,7 +45,7 @@ struct test_layernorm : verify_program<test_layernorm>
{ {
migraphx::program p; migraphx::program p;
auto* mm = p.get_main_module(); auto* mm = p.get_main_module();
std::vector<size_t> dims = {1, 1, 5}; std::vector<int> dims = {1, 1, 5};
auto x = mm->add_parameter("x", migraphx::shape{migraphx::shape::float_type, dims}); auto x = mm->add_parameter("x", migraphx::shape{migraphx::shape::float_type, dims});
add_layernorm(*mm, x, dims); add_layernorm(*mm, x, dims);
return p; return p;
...@@ -58,7 +58,7 @@ struct test_layernorm2 : verify_program<test_layernorm2> ...@@ -58,7 +58,7 @@ struct test_layernorm2 : verify_program<test_layernorm2>
{ {
migraphx::program p; migraphx::program p;
auto* mm = p.get_main_module(); auto* mm = p.get_main_module();
std::vector<size_t> dims = {1, 4, 24}; std::vector<int> dims = {1, 4, 24};
auto x = mm->add_parameter("x", migraphx::shape{migraphx::shape::float_type, dims}); auto x = mm->add_parameter("x", migraphx::shape{migraphx::shape::float_type, dims});
add_layernorm(*mm, x, dims); add_layernorm(*mm, x, dims);
return p; return p;
...@@ -71,7 +71,7 @@ struct test_layernorm_triadd : verify_program<test_layernorm_triadd> ...@@ -71,7 +71,7 @@ struct test_layernorm_triadd : verify_program<test_layernorm_triadd>
{ {
migraphx::program p; migraphx::program p;
auto* mm = p.get_main_module(); auto* mm = p.get_main_module();
std::vector<size_t> dims = {1, 4, 24}; std::vector<int> dims = {1, 4, 24};
auto x = mm->add_parameter("x", migraphx::shape{migraphx::shape::float_type, dims}); auto x = mm->add_parameter("x", migraphx::shape{migraphx::shape::float_type, dims});
auto y = mm->add_parameter("y", migraphx::shape{migraphx::shape::float_type, dims}); auto y = mm->add_parameter("y", migraphx::shape{migraphx::shape::float_type, dims});
auto z = mm->add_parameter("z", migraphx::shape{migraphx::shape::float_type, dims}); auto z = mm->add_parameter("z", migraphx::shape{migraphx::shape::float_type, dims});
...@@ -88,7 +88,7 @@ struct test_layernorm_triadd_large : verify_program<test_layernorm_triadd_large> ...@@ -88,7 +88,7 @@ struct test_layernorm_triadd_large : verify_program<test_layernorm_triadd_large>
{ {
migraphx::program p; migraphx::program p;
auto* mm = p.get_main_module(); auto* mm = p.get_main_module();
std::vector<size_t> dims = {1, 384, 1024}; std::vector<int> dims = {1, 384, 1024};
auto x = mm->add_parameter("x", migraphx::shape{migraphx::shape::float_type, dims}); auto x = mm->add_parameter("x", migraphx::shape{migraphx::shape::float_type, dims});
auto y = mm->add_parameter("y", migraphx::shape{migraphx::shape::float_type, dims}); auto y = mm->add_parameter("y", migraphx::shape{migraphx::shape::float_type, dims});
auto z = mm->add_parameter("z", migraphx::shape{migraphx::shape::float_type, dims}); auto z = mm->add_parameter("z", migraphx::shape{migraphx::shape::float_type, dims});
......
...@@ -12,11 +12,11 @@ struct test_lstm_bidirct_3args : verify_program<test_lstm_bidirct_3args> ...@@ -12,11 +12,11 @@ struct test_lstm_bidirct_3args : verify_program<test_lstm_bidirct_3args>
{ {
migraphx::program create_program() const migraphx::program create_program() const
{ {
std::size_t batch_size = 2; int batch_size = 2;
std::size_t seq_len = 3; int seq_len = 3;
std::size_t hidden_size = 5; int hidden_size = 5;
std::size_t input_size = 8; int input_size = 8;
std::size_t num_dirct = 2; int num_dirct = 2;
float clip = 0.0f; float clip = 0.0f;
migraphx::program p; migraphx::program p;
......
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