Fix wrong descriptor creation logics

1989df75 · Po-Yen, Chen · 3672d070 · 1989df75
Commit 1989df75 authored Sep 08, 2022 by Po-Yen, Chen
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

include/ck/tensor_operation/gpu/device/device_permute.hpp include/ck/tensor_operation/gpu/device/device_permute.hpp +1 -1

No files found.
--- a/include/ck/tensor_operation/gpu/device/device_permute.hpp
+++ b/include/ck/tensor_operation/gpu/device/device_permute.hpp
@@ -176,7 +176,7 @@ struct DevicePermute
            : in_dev_buffer_(static_cast<const InDataType*>(in_dev_buffer)),
              out_dev_buffer_(static_cast<OutDataType*>(out_dev_buffer)),
              in_grid_desc_(MakeDescriptor_N_H_W(inLengths, inStrides)),
-              out_grid_desc_(MakeDescriptor_N_H_W(inLengths, inStrides)),
+              out_grid_desc_(MakeDescriptor_N_H_W(outLengths, outStrides)),
              inLengths_(inLengths),
              inStrides_(inStrides),
              outLengths_(outLengths),