Commit e53b50e8 authored by Po-Yen, Chen's avatar Po-Yen, Chen
Browse files

Create 'DevicePermuteBase' to generate methods

parent ea343345
...@@ -18,6 +18,11 @@ namespace ck { ...@@ -18,6 +18,11 @@ namespace ck {
namespace tensor_operation { namespace tensor_operation {
namespace device { namespace device {
template <typename Derived>
struct DevicePermuteBase : BaseOperator
{
};
template <typename InDataTypeTuple, template <typename InDataTypeTuple,
typename OutDataTypeTuple, typename OutDataTypeTuple,
typename ElementwiseOperation, typename ElementwiseOperation,
...@@ -25,7 +30,13 @@ template <typename InDataTypeTuple, ...@@ -25,7 +30,13 @@ template <typename InDataTypeTuple,
index_t MPerThread, index_t MPerThread,
typename InScalarPerVectorSeq, typename InScalarPerVectorSeq,
typename OutScalarPerVectorSeq> typename OutScalarPerVectorSeq>
struct DevicePermute : BaseOperator struct DevicePermute : DevicePermuteBase<DevicePermute<InDataTypeTuple,
OutDataTypeTuple,
ElementwiseOperation,
NumDim,
MPerThread,
InScalarPerVectorSeq,
OutScalarPerVectorSeq>>
{ {
static constexpr int NumInput = InDataTypeTuple::Size(); static constexpr int NumInput = InDataTypeTuple::Size();
static constexpr int NumOutput = OutDataTypeTuple::Size(); static constexpr int NumOutput = OutDataTypeTuple::Size();
......
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