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
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
......
......@@ -143,6 +143,7 @@ add_library(migraphx_gpu
fuse_ops.cpp
gather.cpp
gemm_impl.cpp
get_tuple_elem.cpp
hip.cpp
int8_conv_pack.cpp
int8_gemm_pack.cpp
......@@ -200,6 +201,7 @@ register_migraphx_gpu_ops(hip_
exp
floor
gather
get_tuple_elem
greater
less
log
......
......@@ -27,6 +27,7 @@
#include <migraphx/gpu/elu.hpp>
#include <migraphx/gpu/equal.hpp>
#include <migraphx/gpu/gemm.hpp>
#include <migraphx/gpu/get_tuple_elem.hpp>
#include <migraphx/gpu/greater.hpp>
#include <migraphx/gpu/int8_conv_pack.hpp>
#include <migraphx/gpu/leaky_relu.hpp>
......@@ -162,6 +163,7 @@ struct miopen_apply
add_extend_op("convert");
add_extend_op("elu");
add_extend_op("gather");
add_extend_op("get_tuple_elem");
add_extend_op("leaky_relu");
add_extend_op("logsoftmax");
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