Change problem description for 'DevicePermute'

692f9e0e · Po-Yen, Chen · bf3ef797 · 692f9e0e
Commit 692f9e0e authored Sep 07, 2022 by Po-Yen, Chen
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 2 deletions

include/ck/tensor_operation/gpu/device/device_permute.hpp include/ck/tensor_operation/gpu/device/device_permute.hpp +4 -2

No files found.
--- a/include/ck/tensor_operation/gpu/device/device_permute.hpp
+++ b/include/ck/tensor_operation/gpu/device/device_permute.hpp
@@ -69,8 +69,10 @@ struct InvokerBase : BaseInvoker
 } // namespace detail

 // Swap last 2 dimensions
-// input: [d0, d1, d2, ..., d, dn-2, dn-1]
-// output: [d0, d1, d2, ..., d, dn-1, dn-2]
+// input shape: [d[0], d[1], d[2], ..., d[NumDim-3], d[NumDim-2], d[NumDim-1]]
+//                                                                ^^^^^^^^^^^
+// output shape: [d[0], d[1], d[2], ..., d[NumDim-3], d[NumDim-1], d[NumDim-2]]
+//                                                    ^^^^^^^^^^^
 template <typename InDataType,
          typename OutDataType,
          typename ElementwiseOperation,