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

Move common parts into common.hpp

parent ccd26cbd
...@@ -16,6 +16,8 @@ ...@@ -16,6 +16,8 @@
#include "ck/library/utility/host_tensor.hpp" #include "ck/library/utility/host_tensor.hpp"
#include "ck/library/utility/host_tensor_generator.hpp" #include "ck/library/utility/host_tensor_generator.hpp"
using F16 = ck::half_t;
struct ExecutionConfig final struct ExecutionConfig final
{ {
bool do_verification = true; bool do_verification = true;
...@@ -28,6 +30,11 @@ struct Problem final ...@@ -28,6 +30,11 @@ struct Problem final
std::array<std::size_t, 4> axes = {0, 2, 3, 1}; std::array<std::size_t, 4> axes = {0, 2, 3, 1};
}; };
template <ck::index_t... Is>
using S = ck::Sequence<Is...>;
using PassThrough = ck::tensor_operation::element_wise::PassThrough;
inline bool parse_cmd_args(int argc, char* argv[], ExecutionConfig& config, Problem& problem) inline bool parse_cmd_args(int argc, char* argv[], ExecutionConfig& config, Problem& problem)
{ {
return true; return true;
......
...@@ -3,20 +3,11 @@ ...@@ -3,20 +3,11 @@
#include "common.hpp" #include "common.hpp"
using F16 = ck::half_t;
using ADataType = F16; using ADataType = F16;
using BDataType = F16; using BDataType = F16;
using PassThrough = ck::tensor_operation::element_wise::PassThrough; using DeviceElementwisePermuteInstance = ck::tensor_operation::device::
using DeviceElementwisePermuteInstance = DeviceElementwise<ck::Tuple<ADataType>, ck::Tuple<BDataType>, PassThrough, 4, 8, S<8>, S<1>>;
ck::tensor_operation::device::DeviceElementwise<ck::Tuple<ADataType>,
ck::Tuple<BDataType>,
PassThrough,
4,
8,
ck::Sequence<8>,
ck::Sequence<1>>;
#include "run_elementwise_permute_example.inc" #include "run_elementwise_permute_example.inc"
......
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