"include/vscode:/vscode.git/clone" did not exist on "e296ee56b35207af047ef3a5cb0f00788c9f2cf0"
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 @@
#include "ck/library/utility/host_tensor.hpp"
#include "ck/library/utility/host_tensor_generator.hpp"
using F16 = ck::half_t;
struct ExecutionConfig final
{
bool do_verification = true;
......@@ -28,6 +30,11 @@ struct Problem final
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)
{
return true;
......
......@@ -3,20 +3,11 @@
#include "common.hpp"
using F16 = ck::half_t;
using ADataType = F16;
using BDataType = F16;
using PassThrough = ck::tensor_operation::element_wise::PassThrough;
using DeviceElementwisePermuteInstance =
ck::tensor_operation::device::DeviceElementwise<ck::Tuple<ADataType>,
ck::Tuple<BDataType>,
PassThrough,
4,
8,
ck::Sequence<8>,
ck::Sequence<1>>;
using DeviceElementwisePermuteInstance = ck::tensor_operation::device::
DeviceElementwise<ck::Tuple<ADataType>, ck::Tuple<BDataType>, PassThrough, 4, 8, S<8>, S<1>>;
#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