Skip to content

GitLab

"...composable_kernel.git" did not exist on "7bcaf2a70f3f297c27cdc59a58f44f777837465b"
Commit 692f9e0e authored by Po-Yen, Chen's avatar Po-Yen, Chen
Browse files

Change problem description for 'DevicePermute'

parent bf3ef797
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 2 deletions
include/ck/tensor_operation/gpu/device/device_permute.hpp
View file @ 692f9e0e
...@@ -69,8 +69,10 @@ struct InvokerBase : BaseInvoker ...@@ -69,8 +69,10 @@ struct InvokerBase : BaseInvoker
} // namespace detail } // namespace detail
// Swap last 2 dimensions // Swap last 2 dimensions
// input: [d0, d1, d2, ..., d, dn-2, dn-1] // input shape: [d[0], d[1], d[2], ..., d[NumDim-3], d[NumDim-2], d[NumDim-1]]
// output: [d0, d1, d2, ..., d, dn-1, dn-2] // ^^^^^^^^^^^
// output shape: [d[0], d[1], d[2], ..., d[NumDim-3], d[NumDim-1], d[NumDim-2]]
// ^^^^^^^^^^^
template <typename InDataType, template <typename InDataType,
typename OutDataType, typename OutDataType,
typename ElementwiseOperation, typename ElementwiseOperation,
......
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