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