Skip to content

GitLab

Commit a822937a authored by ltqin's avatar ltqin
Browse files

change IsTileSkippable parameters name

parent b0ae034e
Show whitespace changes
Inline Side-by-side
Showing with 5 additions and 3 deletions
include/ck/tensor_operation/gpu/device/masking_specialization.hpp
View file @ a822937a
...@@ -69,10 +69,12 @@ struct MaskUpperTriangleFromBottomRightPredicate ...@@ -69,10 +69,12 @@ struct MaskUpperTriangleFromBottomRightPredicate
return n > (m + diagonal_offset_); return n > (m + diagonal_offset_);
} }
__host__ __device__ constexpr bool __host__ __device__ constexpr bool IsTileSkippable(index_t m_tile_orig,
IsTileSkippable(index_t m, index_t n, index_t m_tile, index_t /*n_tile*/) const 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: private:
......
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