Skip to content

GitLab

"...composable_kernel.git" did not exist on "80da57fd320d12c2f470197ffcbba61ffe1ad053"
Commit c7aa455f authored by Po-Yen, Chen's avatar Po-Yen, Chen
Browse files

Re-format 'DeviceElementwise'

parent 339e51d1
Hide whitespace changes
Inline Side-by-side
Showing with 8 additions and 2 deletions
include/ck/tensor_operation/gpu/device/device_elementwise.hpp
View file @ c7aa455f
...@@ -260,14 +260,20 @@ struct DeviceElementwise ...@@ -260,14 +260,20 @@ struct DeviceElementwise
return IsSupportedArgument(*dynamic_cast<const Argument*>(p_arg)); return IsSupportedArgument(*dynamic_cast<const Argument*>(p_arg));
} }
static auto MakeArgument(const std::array<index_t, NumDim> lengths, static auto
MakeArgument(const std::array<index_t, NumDim> lengths,
const std::array<std::array<index_t, NumDim>, NumInput> inStridesArray, const std::array<std::array<index_t, NumDim>, NumInput> inStridesArray,
const std::array<std::array<index_t, NumDim>, NumOutput> outStridesArray, const std::array<std::array<index_t, NumDim>, NumOutput> outStridesArray,
const std::array<const void*, NumInput> in_dev_buffers, const std::array<const void*, NumInput> in_dev_buffers,
const std::array<void*, NumOutput> out_dev_buffers, const std::array<void*, NumOutput> out_dev_buffers,
ElementwiseOperation elementwise_op) ElementwiseOperation elementwise_op)
{ {
return Argument{lengths, inStridesArray, outStridesArray, in_dev_buffers, out_dev_buffers, elementwise_op}; return Argument{lengths,
inStridesArray,
outStridesArray,
in_dev_buffers,
out_dev_buffers,
elementwise_op};
} }
std::unique_ptr<BaseArgument> std::unique_ptr<BaseArgument>
......
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