"megatron/vscode:/vscode.git/clone" did not exist on "d928dd5159ff95a380f8169e43b06d69abe006ec"
Commit 79c9c44d authored by Shucai Xiao's avatar Shucai Xiao
Browse files

add missing files for get_tuple_elem

parent c57b18ea
#include <migraphx/gpu/get_tuple_elem.hpp>
#include <migraphx/gpu/context.hpp>
#include <migraphx/gpu/device/contiguous.hpp>
namespace migraphx {
inline namespace MIGRAPHX_INLINE_NS {
namespace gpu {
shape hip_get_tuple_elem::compute_shape(std::vector<shape> inputs) const
{
inputs.pop_back();
return op.compute_shape(inputs);
}
argument hip_get_tuple_elem::compute(context& ctx, const shape&, const std::vector<argument>& args) const
{
auto sub_args = args.front().get_sub_objects();
device::contiguous(ctx.get_stream().get(), args.back(), sub_args[op.index]);
return args.back();
}
} // namespace gpu
} // namespace MIGRAPHX_INLINE_NS
} // namespace migraphx
#ifndef MIGRAPHX_GUARD_RTGLIB_GET_TUPLE_ELEM_HPP
#define MIGRAPHX_GUARD_RTGLIB_GET_TUPLE_ELEM_HPP
#include <migraphx/argument.hpp>
#include <migraphx/reflect.hpp>
#include <migraphx/op/get_tuple_elem.hpp>
#include <migraphx/gpu/miopen.hpp>
namespace migraphx {
inline namespace MIGRAPHX_INLINE_NS {
namespace gpu {
struct context;
struct hip_get_tuple_elem
{
op::get_tuple_elem op;
template <class Self, class F>
static auto reflect(Self& self, F f)
{
return migraphx::reflect(self.op, f);
}
std::string name() const { return "gpu::get_tuple_elem"; }
shape compute_shape(std::vector<shape> inputs) const;
argument
compute(context& ctx, const shape& output_shape, const std::vector<argument>& args) const;
std::ptrdiff_t output_alias(const std::vector<shape>& shapes) const
{
return shapes.size() - 1;
}
};
} // namespace gpu
} // namespace MIGRAPHX_INLINE_NS
} // namespace migraphx
#endif
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