add comments for MaskUpperTriangleFromBottomRightPredicate

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

Showing with 7 additions and 0 deletions

include/ck/tensor_operation/gpu/device/masking_specialization.hpp ...ck/tensor_operation/gpu/device/masking_specialization.hpp +7 -0

No files found.
--- a/include/ck/tensor_operation/gpu/device/masking_specialization.hpp
+++ b/include/ck/tensor_operation/gpu/device/masking_specialization.hpp
@@ -50,6 +50,13 @@ struct MaskUpperTriangleFromTopLeftPredicate
        return operator()(m + m_tile - 1, n);
    }
 };
+
+// eg: m = 3, n = 5 => offset = 2
+//    so matrix(n > m + offset) = 0
+//      1  2  3  4  5
+//    1 *  *  *  0  0
+//    2 *  *  *  *  0
+//    3 *  *  *  *  *
 struct MaskUpperTriangleFromBottomRightPredicate
 {
    MaskUpperTriangleFromBottomRightPredicate() : diagonal_offset_(0) {}