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

Remove template BaseInvokerCRTP<>

parent 1ca0b97c
......@@ -57,24 +57,6 @@ struct BaseOperator
virtual ~BaseOperator() {}
};
template <typename DerivedInvoker, typename Argument>
struct BaseInvokerCRTP : BaseInvoker
{
static_assert(std::is_class_v<Argument> && std::is_base_of_v<BaseArgument, Argument>);
float Run(const BaseArgument* arg,
const StreamConfig& stream_config = StreamConfig{}) override final
{
const auto* const argument = dynamic_cast<const Argument*>(arg);
if(!argument)
{
return NAN;
}
return DerivedInvoker::Run(*argument, stream_config);
}
};
} // namespace device
} // namespace tensor_operation
} // namespace ck
......@@ -146,7 +146,7 @@ struct DevicePermuteImpl : DevicePermute<NumDim, InDataType, OutDataType, Elemen
Block2TileMap block_2_tile_map_;
};
struct Invoker : BaseInvokerCRTP<Invoker, Argument>
struct Invoker : BaseInvoker
{
static float Run(const Argument& arg, const StreamConfig& stream_config = StreamConfig{})
{
......@@ -173,6 +173,18 @@ struct DevicePermuteImpl : DevicePermute<NumDim, InDataType, OutDataType, Elemen
arg.block_2_tile_map_);
return elapsed_time;
}
float Run(const BaseArgument* arg,
const StreamConfig& stream_config = StreamConfig{}) override final
{
const auto* const argument = dynamic_cast<const Argument*>(arg);
if(!argument)
{
return NAN;
}
return Run(*argument, stream_config);
}
};
static bool IsSupportedArgument(const Argument& arg)
......
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