"...composable_kernel_rocm.git" did not exist on "b94fd0b2279c6476c6e109e99dc5d0e6d8ce313c"
Commit f015e568 authored by Po-Yen, Chen's avatar Po-Yen, Chen
Browse files

Remove base class of 'DevicePermute'

parent 6c4268f9
...@@ -26,7 +26,6 @@ template <typename InDataTypeTuple, ...@@ -26,7 +26,6 @@ template <typename InDataTypeTuple,
typename InScalarPerVectorSeq, typename InScalarPerVectorSeq,
typename OutScalarPerVectorSeq> typename OutScalarPerVectorSeq>
struct DevicePermute struct DevicePermute
: public DeviceElementwiseBase<InDataTypeTuple, OutDataTypeTuple, ElementwiseOperation, NumDim>
{ {
static constexpr int NumInput = InDataTypeTuple::Size(); static constexpr int NumInput = InDataTypeTuple::Size();
static constexpr int NumOutput = OutDataTypeTuple::Size(); static constexpr int NumOutput = OutDataTypeTuple::Size();
...@@ -255,7 +254,7 @@ struct DevicePermute ...@@ -255,7 +254,7 @@ struct DevicePermute
return valid; return valid;
}; };
bool IsSupportedArgument(const BaseArgument* p_arg) override bool IsSupportedArgument(const BaseArgument* p_arg)
{ {
return IsSupportedArgument(*dynamic_cast<const Argument*>(p_arg)); return IsSupportedArgument(*dynamic_cast<const Argument*>(p_arg));
} }
...@@ -282,7 +281,7 @@ struct DevicePermute ...@@ -282,7 +281,7 @@ struct DevicePermute
const std::array<std::array<index_t, NumDim>, NumOutput> outStridesArray, const std::array<std::array<index_t, NumDim>, NumOutput> outStridesArray,
const std::array<const void*, NumInput> in_dev_buffers, const std::array<const void*, NumInput> in_dev_buffers,
const std::array<void*, NumOutput> out_dev_buffers, const std::array<void*, NumOutput> out_dev_buffers,
ElementwiseOperation elementwise_op) override ElementwiseOperation elementwise_op)
{ {
return std::make_unique<Argument>(lengths, return std::make_unique<Argument>(lengths,
inStridesArray, inStridesArray,
...@@ -293,10 +292,7 @@ struct DevicePermute ...@@ -293,10 +292,7 @@ struct DevicePermute
} }
static auto MakeInvoker() { return Invoker{}; } static auto MakeInvoker() { return Invoker{}; }
std::unique_ptr<BaseInvoker> MakeInvokerPointer() override std::unique_ptr<BaseInvoker> MakeInvokerPointer() { return std::make_unique<Invoker>(); };
{
return std::make_unique<Invoker>();
};
}; // namespace device }; // namespace device
} // namespace device } // namespace device
......
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