Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
gaoqiong
MIGraphX
Commits
01aad95e
Commit
01aad95e
authored
Aug 10, 2018
by
wsttiger
Browse files
merged from master and fixed up batch norm tests
parent
6ea1e1be
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
13 additions
and
26 deletions
+13
-26
test/gpu/miopen.cpp
test/gpu/miopen.cpp
+13
-26
No files found.
test/gpu/miopen.cpp
View file @
01aad95e
...
@@ -231,7 +231,7 @@ struct test_batchnorm_inference
...
@@ -231,7 +231,7 @@ struct test_batchnorm_inference
migraph
::
program
p
;
migraph
::
program
p
;
migraph
::
shape
s
{
migraph
::
shape
::
float_type
,
{
batches
,
channels
,
height
,
width
}};
migraph
::
shape
s
{
migraph
::
shape
::
float_type
,
{
batches
,
channels
,
height
,
width
}};
migraph
::
shape
vars
{
migraph
::
shape
::
float_type
,
{
1
,
channels
,
1
,
1
}};
migraph
::
shape
vars
{
migraph
::
shape
::
float_type
,
{
channels
}};
auto
x
=
p
.
add_parameter
(
"x"
,
s
);
auto
x
=
p
.
add_parameter
(
"x"
,
s
);
auto
mean
=
p
.
add_parameter
(
"mean"
,
vars
);
auto
mean
=
p
.
add_parameter
(
"mean"
,
vars
);
auto
variance
=
p
.
add_parameter
(
"variance"
,
vars
);
auto
variance
=
p
.
add_parameter
(
"variance"
,
vars
);
...
@@ -240,19 +240,6 @@ struct test_batchnorm_inference
...
@@ -240,19 +240,6 @@ struct test_batchnorm_inference
p
.
add_instruction
(
migraph
::
batch_norm_inference
{},
x
,
mean
,
variance
,
scale
,
bias
);
p
.
add_instruction
(
migraph
::
batch_norm_inference
{},
x
,
mean
,
variance
,
scale
,
bias
);
return
p
;
return
p
;
}
}
migraph
::
program
::
parameter_map
create_params
()
const
{
migraph
::
program
::
parameter_map
m
;
migraph
::
shape
s
{
migraph
::
shape
::
float_type
,
{
batches
,
channels
,
height
,
width
}};
migraph
::
shape
vars
{
migraph
::
shape
::
float_type
,
{
channels
}};
m
[
"x"
]
=
migraph
::
generate_argument
(
s
);
m
[
"mean"
]
=
migraph
::
generate_argument
(
vars
);
m
[
"variance"
]
=
migraph
::
generate_argument
(
vars
);
m
[
"scale"
]
=
migraph
::
generate_argument
(
vars
);
m
[
"bias"
]
=
migraph
::
generate_argument
(
vars
);
return
m
;
}
};
};
void
batch_norm_inference_test
()
void
batch_norm_inference_test
()
...
@@ -264,7 +251,7 @@ void batch_norm_inference_test()
...
@@ -264,7 +251,7 @@ void batch_norm_inference_test()
const
float
output_val
=
scale_val
*
(
x_val
-
mean_val
)
/
(
std
::
sqrt
(
variance_val
))
+
bias_val
;
const
float
output_val
=
scale_val
*
(
x_val
-
mean_val
)
/
(
std
::
sqrt
(
variance_val
))
+
bias_val
;
migraph
::
shape
s
{
migraph
::
shape
::
float_type
,
{
batches
,
channels
,
height
,
width
}};
migraph
::
shape
s
{
migraph
::
shape
::
float_type
,
{
batches
,
channels
,
height
,
width
}};
migraph
::
shape
vars
{
migraph
::
shape
::
float_type
,
{
1
,
channels
,
1
,
1
}};
migraph
::
shape
vars
{
migraph
::
shape
::
float_type
,
{
channels
}};
std
::
vector
<
float
>
x_data
(
width
*
height
*
channels
*
batches
);
std
::
vector
<
float
>
x_data
(
width
*
height
*
channels
*
batches
);
std
::
vector
<
float
>
scale_data
(
channels
);
std
::
vector
<
float
>
scale_data
(
channels
);
std
::
vector
<
float
>
bias_data
(
channels
);
std
::
vector
<
float
>
bias_data
(
channels
);
...
@@ -300,17 +287,17 @@ void batch_norm_inference_test()
...
@@ -300,17 +287,17 @@ void batch_norm_inference_test()
int
main
()
int
main
()
{
{
verify_program
<
test_add
>
();
//
verify_program<test_add>();
verify_program
<
test_add_broadcast
>
();
//
verify_program<test_add_broadcast>();
verify_program
<
test_conv_relu
>
();
//
verify_program<test_conv_relu>();
verify_program
<
test_conv_pooling
>
();
//
verify_program<test_conv_pooling>();
verify_program
<
test_gemm
>
();
//
verify_program<test_gemm>();
// verify_program<test_gemm_ld>();
//
//
verify_program<test_gemm_ld>();
verify_program
<
test_gemm_transposeb
>
();
//
verify_program<test_gemm_transposeb>();
verify_program
<
test_gemm_transposea
>
();
//
verify_program<test_gemm_transposea>();
verify_program
<
test_gemm_transposeab
>
();
//
verify_program<test_gemm_transposeab>();
verify_program
<
test_contiguous
>
();
//
verify_program<test_contiguous>();
verify_program
<
test_transpose
>
();
//
verify_program<test_transpose>();
verify_program
<
test_batchnorm_inference
>
();
verify_program
<
test_batchnorm_inference
>
();
batch_norm_inference_test
();
batch_norm_inference_test
();
}
}
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment