Skip to content

GitLab

Commit 59fef16f authored by Po-Yen, Chen's avatar Po-Yen, Chen
Browse files

Add comment

parent 360383bb
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
include/ck/tensor_operation/gpu/grid/gridwise_permute.hpp
View file @ 59fef16f
...@@ -169,6 +169,7 @@ struct GridwisePermute ...@@ -169,6 +169,7 @@ struct GridwisePermute
{ {
constexpr index_t NumDim = InGridDesc::GetNumOfDimension(); constexpr index_t NumDim = InGridDesc::GetNumOfDimension();
// check if we only swap last 2 dimensions
bool valid = true; bool valid = true;
static_for<0, NumDim - 2, 1>{}([&](auto I) { static_for<0, NumDim - 2, 1>{}([&](auto I) {
if(valid && in_grid_desc.GetLength(I) != out_grid_desc.GetLength(I)) if(valid && in_grid_desc.GetLength(I) != out_grid_desc.GetLength(I))
......
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