Let 'DevicePermute' inherit from 'BaseOperator'

bc26a2fa · Po-Yen, Chen · f015e568 · bc26a2fa
Commit bc26a2fa authored Sep 06, 2022 by Po-Yen, Chen
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

include/ck/tensor_operation/gpu/device/device_permute.hpp include/ck/tensor_operation/gpu/device/device_permute.hpp +2 -2

No files found.
--- a/include/ck/tensor_operation/gpu/device/device_permute.hpp
+++ b/include/ck/tensor_operation/gpu/device/device_permute.hpp
@@ -25,7 +25,7 @@ template <typename InDataTypeTuple,
          index_t MPerThread,
          typename InScalarPerVectorSeq,
          typename OutScalarPerVectorSeq>
-struct DevicePermute
+struct DevicePermute : BaseOperator
 {
    static constexpr int NumInput  = InDataTypeTuple::Size();
    static constexpr int NumOutput = OutDataTypeTuple::Size();
@@ -254,7 +254,7 @@ struct DevicePermute
        return valid;
    };

-    bool IsSupportedArgument(const BaseArgument* p_arg)
+    bool IsSupportedArgument(const BaseArgument* p_arg) override
    {
        return IsSupportedArgument(*dynamic_cast<const Argument*>(p_arg));
    }