Commit ce553b38 authored by Khalique's avatar Khalique
Browse files

fixed gpu lowering implementation

parent 7c453f85
...@@ -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
......
...@@ -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));
......
...@@ -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>();
......
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