Commit c57b18ea authored by Shucai Xiao's avatar Shucai Xiao
Browse files

add a gpu implementation for the get_tuple_elem op

parent fc42d852
...@@ -46,7 +46,7 @@ struct get_tuple_elem ...@@ -46,7 +46,7 @@ struct get_tuple_elem
return vec_args.at(index); return vec_args.at(index);
} }
std::ptrdiff_t output_alias(const std::vector<shape>&) const { return 0; } // std::ptrdiff_t output_alias(const std::vector<shape>&) const { return 0; }
}; };
} // namespace op } // namespace op
......
...@@ -143,6 +143,7 @@ add_library(migraphx_gpu ...@@ -143,6 +143,7 @@ add_library(migraphx_gpu
fuse_ops.cpp fuse_ops.cpp
gather.cpp gather.cpp
gemm_impl.cpp gemm_impl.cpp
get_tuple_elem.cpp
hip.cpp hip.cpp
int8_conv_pack.cpp int8_conv_pack.cpp
int8_gemm_pack.cpp int8_gemm_pack.cpp
...@@ -200,6 +201,7 @@ register_migraphx_gpu_ops(hip_ ...@@ -200,6 +201,7 @@ register_migraphx_gpu_ops(hip_
exp exp
floor floor
gather gather
get_tuple_elem
greater greater
less less
log log
......
...@@ -27,6 +27,7 @@ ...@@ -27,6 +27,7 @@
#include <migraphx/gpu/elu.hpp> #include <migraphx/gpu/elu.hpp>
#include <migraphx/gpu/equal.hpp> #include <migraphx/gpu/equal.hpp>
#include <migraphx/gpu/gemm.hpp> #include <migraphx/gpu/gemm.hpp>
#include <migraphx/gpu/get_tuple_elem.hpp>
#include <migraphx/gpu/greater.hpp> #include <migraphx/gpu/greater.hpp>
#include <migraphx/gpu/int8_conv_pack.hpp> #include <migraphx/gpu/int8_conv_pack.hpp>
#include <migraphx/gpu/leaky_relu.hpp> #include <migraphx/gpu/leaky_relu.hpp>
...@@ -162,6 +163,7 @@ struct miopen_apply ...@@ -162,6 +163,7 @@ struct miopen_apply
add_extend_op("convert"); add_extend_op("convert");
add_extend_op("elu"); add_extend_op("elu");
add_extend_op("gather"); add_extend_op("gather");
add_extend_op("get_tuple_elem");
add_extend_op("leaky_relu"); add_extend_op("leaky_relu");
add_extend_op("logsoftmax"); add_extend_op("logsoftmax");
add_extend_op("lrn"); add_extend_op("lrn");
......
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