Release the constraint on In/OutGridDesc

62d7361f · Po-Yen, Chen · 9f734b3c · 62d7361f
Commit 62d7361f authored Sep 08, 2022 by Po-Yen, Chen
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

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

No files found.
--- a/include/ck/tensor_operation/gpu/grid/gridwise_permute.hpp
+++ b/include/ck/tensor_operation/gpu/grid/gridwise_permute.hpp
@@ -104,7 +104,8 @@ template <typename InGridDesc,
          index_t OutScalarPerVector>
 struct GridwisePermute
 {
-    static_assert(InGridDesc::GetNumOfDimension() == 3 && OutGridDesc::GetNumOfDimension() == 3);
+    static_assert(InGridDesc::GetNumOfDimension() == OutGridDesc::GetNumOfDimension());
+    static_assert(3 <= InGridDesc::GetNumOfDimension());

    static constexpr auto I0 = Number<0>{};
    static constexpr auto I1 = Number<1>{};