Create 'DevicePermuteBase' to generate methods

e53b50e8 · Po-Yen, Chen · ea343345 · e53b50e8
Commit e53b50e8 authored Sep 06, 2022 by Po-Yen, Chen
Hide whitespace changes
Inline Side-by-side

Showing with 12 additions and 1 deletion

include/ck/tensor_operation/gpu/device/device_permute.hpp include/ck/tensor_operation/gpu/device/device_permute.hpp +12 -1

No files found.
--- a/include/ck/tensor_operation/gpu/device/device_permute.hpp
+++ b/include/ck/tensor_operation/gpu/device/device_permute.hpp
@@ -18,6 +18,11 @@ namespace ck {
 namespace tensor_operation {
 namespace device {

+template <typename Derived>
+struct DevicePermuteBase : BaseOperator
+{
+};
+
 template <typename InDataTypeTuple,
          typename OutDataTypeTuple,
          typename ElementwiseOperation,
@@ -25,7 +30,13 @@ template <typename InDataTypeTuple,
          index_t MPerThread,
          typename InScalarPerVectorSeq,
          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 NumOutput = OutDataTypeTuple::Size();