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

change size_t to int

parent 26bd92d8
......@@ -12,11 +12,11 @@ struct test_rnn_3args : verify_program<test_rnn_3args>
{
migraphx::program create_program() const
{
std::size_t batch_size = 2;
std::size_t seq_len = 1;
std::size_t hidden_size = 4;
std::size_t input_size = 3;
std::size_t num_dirct = 1;
int batch_size = 2;
int seq_len = 1;
int hidden_size = 4;
int input_size = 3;
int num_dirct = 1;
float clip = 0.0f;
migraphx::program p;
......
......@@ -12,11 +12,11 @@ struct test_rnn_4args : verify_program<test_rnn_4args>
{
migraphx::program create_program() const
{
std::size_t batch_size = 2;
std::size_t seq_len = 5;
std::size_t hidden_size = 4;
std::size_t input_size = 3;
std::size_t num_dirct = 1;
int batch_size = 2;
int seq_len = 5;
int hidden_size = 4;
int input_size = 3;
int num_dirct = 1;
float clip = 0.0f;
migraphx::program p;
......
......@@ -12,11 +12,11 @@ struct test_rnn_5args : verify_program<test_rnn_5args>
{
migraphx::program create_program() const
{
std::size_t batch_size = 2;
std::size_t seq_len = 10;
std::size_t hidden_size = 4;
std::size_t input_size = 3;
std::size_t num_dirct = 1;
int batch_size = 2;
int seq_len = 10;
int hidden_size = 4;
int input_size = 3;
int num_dirct = 1;
float clip = 0.0f;
migraphx::program p;
......
......@@ -12,11 +12,11 @@ struct test_rnn_bi_3args : verify_program<test_rnn_bi_3args>
{
migraphx::program create_program() const
{
std::size_t batch_size = 2;
std::size_t seq_len = 10;
std::size_t hidden_size = 4;
std::size_t input_size = 3;
std::size_t num_dirct = 2;
int batch_size = 2;
int seq_len = 10;
int hidden_size = 4;
int input_size = 3;
int num_dirct = 2;
float clip = 0.0f;
migraphx::program p;
......
......@@ -12,11 +12,11 @@ struct test_rnn_bidirectional : verify_program<test_rnn_bidirectional>
{
migraphx::program create_program() const
{
std::size_t batch_size = 2;
std::size_t seq_len = 1;
std::size_t hidden_size = 4;
std::size_t input_size = 3;
std::size_t num_dirct = 2;
int batch_size = 2;
int seq_len = 1;
int hidden_size = 4;
int input_size = 3;
int num_dirct = 2;
float clip = 0.0f;
migraphx::program p;
......
......@@ -12,11 +12,11 @@ struct test_rnn_bidirectional10 : verify_program<test_rnn_bidirectional10>
{
migraphx::program create_program() const
{
std::size_t batch_size = 2;
std::size_t seq_len = 10;
std::size_t hidden_size = 4;
std::size_t input_size = 3;
std::size_t num_dirct = 2;
int batch_size = 2;
int seq_len = 10;
int hidden_size = 4;
int input_size = 3;
int num_dirct = 2;
float clip = 0.0f;
migraphx::program p;
......
......@@ -12,11 +12,11 @@ struct test_rnn_forward : verify_program<test_rnn_forward>
{
migraphx::program create_program() const
{
std::size_t batch_size = 2;
std::size_t seq_len = 1;
std::size_t hidden_size = 4;
std::size_t input_size = 3;
std::size_t num_dirct = 1;
int batch_size = 2;
int seq_len = 1;
int hidden_size = 4;
int input_size = 3;
int num_dirct = 1;
float clip = 0.0f;
migraphx::program p;
......
......@@ -12,11 +12,11 @@ struct test_rnn_forward10 : verify_program<test_rnn_forward10>
{
migraphx::program create_program() const
{
std::size_t batch_size = 2;
std::size_t seq_len = 10;
std::size_t hidden_size = 4;
std::size_t input_size = 3;
std::size_t num_dirct = 1;
int batch_size = 2;
int seq_len = 10;
int hidden_size = 4;
int input_size = 3;
int num_dirct = 1;
float clip = 0.0f;
migraphx::program p;
......
......@@ -12,11 +12,11 @@ struct test_rnn_reverse : verify_program<test_rnn_reverse>
{
migraphx::program create_program() const
{
std::size_t batch_size = 2;
std::size_t seq_len = 1;
std::size_t hidden_size = 4;
std::size_t input_size = 3;
std::size_t num_dirct = 1;
int batch_size = 2;
int seq_len = 1;
int hidden_size = 4;
int input_size = 3;
int num_dirct = 1;
float clip = 0.0f;
migraphx::program p;
......
......@@ -12,11 +12,11 @@ struct test_rnn_reverse2 : verify_program<test_rnn_reverse2>
{
migraphx::program create_program() const
{
std::size_t batch_size = 2;
std::size_t seq_len = 2;
std::size_t hidden_size = 4;
std::size_t input_size = 3;
std::size_t num_dirct = 1;
int batch_size = 2;
int seq_len = 2;
int hidden_size = 4;
int input_size = 3;
int num_dirct = 1;
float clip = 0.0f;
migraphx::program p;
......
......@@ -12,11 +12,11 @@ struct test_rnn_sql_1 : verify_program<test_rnn_sql_1>
{
migraphx::program create_program() const
{
std::size_t batch_size = 2;
std::size_t seq_len = 10;
std::size_t hidden_size = 4;
std::size_t input_size = 3;
std::size_t num_dirct = 1;
int batch_size = 2;
int seq_len = 10;
int hidden_size = 4;
int input_size = 3;
int num_dirct = 1;
float clip = 0.0f;
migraphx::program p;
......
......@@ -12,11 +12,11 @@ struct test_rnn_sql_2 : verify_program<test_rnn_sql_2>
{
migraphx::program create_program() const
{
std::size_t batch_size = 2;
std::size_t seq_len = 10;
std::size_t hidden_size = 4;
std::size_t input_size = 3;
std::size_t num_dirct = 1;
int batch_size = 2;
int seq_len = 10;
int hidden_size = 4;
int input_size = 3;
int num_dirct = 1;
float clip = 0.0f;
migraphx::program p;
......
......@@ -10,7 +10,7 @@ struct test_rsqrt : verify_program<test_rsqrt>
{
migraphx::program p;
auto* mm = p.get_main_module();
std::vector<size_t> input_lens{1, 3, 16, 16};
std::vector<int> input_lens{1, 3, 16, 16};
migraphx::shape s{migraphx::shape::float_type, input_lens};
auto x = mm->add_parameter("x", s);
auto min_val = mm->add_literal(1.0f);
......
......@@ -12,11 +12,11 @@ struct test_var_sl_gru_bidirct : verify_program<test_var_sl_gru_bidirct>
{
migraphx::program create_program() const
{
std::size_t batch_size = 3;
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 = 3;
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_var_sl_gru_forward : verify_program<test_var_sl_gru_forward>
{
migraphx::program create_program() const
{
std::size_t batch_size = 3;
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 = 3;
int seq_len = 3;
int hidden_size = 5;
int input_size = 8;
int num_dirct = 1;
float clip = 0.0f;
migraphx::program p;
......
......@@ -217,9 +217,9 @@ class Parameter:
self.size_cparam = len(self.cparams)
self.size_name = name or self.name + '_size'
if self.returns:
self.add_param('size_t *', self.size_name)
self.add_param('int *', self.size_name)
else:
self.add_param('size_t', self.size_name)
self.add_param('int', self.size_name)
def bad_param(self, cond: str, msg: str) -> None:
self.bad_param_check = BadParam(cond, msg)
......@@ -325,7 +325,7 @@ class Function:
def share_params(self) -> None:
if self.shared_size == True:
size_param_name = 'size'
size_type = Type('size_t')
size_type = Type('int')
for param in self.params:
p = param.remove_size_param(size_param_name)
if p:
......@@ -796,7 +796,7 @@ def string_c_wrap(p: Parameter) -> None:
p.write = ['*${name} = ${result}.c_str()']
else:
p.add_param(t)
p.add_param('size_t', p.name + '_size')
p.add_param('int', p.name + '_size')
p.bad_param('${name} == nullptr', 'Null pointer')
p.cpp_write = '${type}(${name})'
p.write = [
......
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