change IsTileSkippable parameters name

a822937a · ltqin · b0ae034e · a822937a
Commit a822937a authored Jul 27, 2023 by ltqin
Show whitespace changes
Inline Side-by-side

Showing with 5 additions and 3 deletions

include/ck/tensor_operation/gpu/device/masking_specialization.hpp ...ck/tensor_operation/gpu/device/masking_specialization.hpp +5 -3

No files found.
--- a/include/ck/tensor_operation/gpu/device/masking_specialization.hpp
+++ b/include/ck/tensor_operation/gpu/device/masking_specialization.hpp
@@ -69,10 +69,12 @@ struct MaskUpperTriangleFromBottomRightPredicate
        return n > (m + diagonal_offset_);
    }

-    __host__ __device__ constexpr bool
-    IsTileSkippable(index_t m, index_t n, index_t m_tile, index_t /*n_tile*/) const
+    __host__ __device__ constexpr bool IsTileSkippable(index_t m_tile_orig,
+                                                       index_t n_tile_orig,
+                                                       index_t m_tile_size,
+                                                       index_t /*n_tile_size*/) const
    {
-        return operator()(m + m_tile - 1, n);
+        return operator()(m_tile_orig + m_tile_size - 1, n_tile_orig);
    }

    private: