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
ce553b38
Commit
ce553b38
authored
Oct 22, 2018
by
Khalique
Browse files
fixed gpu lowering implementation
parent
7c453f85
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
6 additions
and
1 deletion
+6
-1
src/targets/gpu/CMakeLists.txt
src/targets/gpu/CMakeLists.txt
+1
-0
src/targets/gpu/lowering.cpp
src/targets/gpu/lowering.cpp
+4
-0
test/gpu/miopen.cpp
test/gpu/miopen.cpp
+1
-1
No files found.
src/targets/gpu/CMakeLists.txt
View file @
ce553b38
...
@@ -32,6 +32,7 @@ add_library(migraph_gpu
...
@@ -32,6 +32,7 @@ add_library(migraph_gpu
softmax.cpp
softmax.cpp
contiguous.cpp
contiguous.cpp
relu.cpp
relu.cpp
leaky_relu.cpp
add.cpp
add.cpp
batchnorm.cpp
batchnorm.cpp
write_literals.cpp
write_literals.cpp
...
...
src/targets/gpu/lowering.cpp
View file @
ce553b38
...
@@ -52,6 +52,10 @@ struct miopen_apply
...
@@ -52,6 +52,10 @@ struct miopen_apply
{
{
check_shape
(
s
,
apply_activation
(
it
));
check_shape
(
s
,
apply_activation
(
it
));
}
}
else
if
(
it
->
name
()
==
"leaky_relu"
)
{
check_shape
(
s
,
apply_leaky_relu
(
it
));
}
else
if
(
it
->
name
()
==
"pooling"
)
else
if
(
it
->
name
()
==
"pooling"
)
{
{
check_shape
(
s
,
apply_pooling
(
it
));
check_shape
(
s
,
apply_pooling
(
it
));
...
...
test/gpu/miopen.cpp
View file @
ce553b38
...
@@ -633,7 +633,7 @@ int main()
...
@@ -633,7 +633,7 @@ int main()
verify_program
<
test_leaky_relu
>
();
verify_program
<
test_leaky_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
>
();
...
...
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