Let 'DstVectorDim' equals 'SrcVectorDim' after transpose out grid desc

ac9d0a67 · Po-Yen, Chen · c3943745 · ac9d0a67
Commit ac9d0a67 authored Sep 08, 2022 by Po-Yen, Chen
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 2 deletions

include/ck/tensor_operation/gpu/grid/gridwise_permute.hpp include/ck/tensor_operation/gpu/grid/gridwise_permute.hpp +1 -2

No files found.
--- a/include/ck/tensor_operation/gpu/grid/gridwise_permute.hpp
+++ b/include/ck/tensor_operation/gpu/grid/gridwise_permute.hpp
@@ -221,8 +221,7 @@ struct GridwisePermute

        constexpr index_t SrcVectorDimAfterMerge =
            SrcVectorDim - (InGridDesc::GetNumOfDimension() - 3);
-        constexpr index_t DstVectorDimAfterMerge =
-            DstVectorDim - (OutGridDesc::GetNumOfDimension() - 3);
+        constexpr index_t DstVectorDimAfterMerge = SrcVectorDimAfterMerge;

        using ck::tensor_operation::element_wise::PassThrough;