From 3f076a6fc147362c67f0540f84b465ac3b906a71 Mon Sep 17 00:00:00 2001
From: root <root@cv350-ccs-aus-h15-18.cs-aus.dcgpu>
Date: Thu, 23 Apr 2026 08:17:34 +0000
Subject: [PATCH] Add IsOutOfSinkBound alias in GenericAttentionMask (API
 compatibility)

Made-with: Cursor
---
 .../ck_tile/ops/unified_attention/block/block_masking.hpp   | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/include/ck_tile/ops/unified_attention/block/block_masking.hpp b/include/ck_tile/ops/unified_attention/block/block_masking.hpp
index 68eb6efc89..c2384511ce 100644
--- a/include/ck_tile/ops/unified_attention/block/block_masking.hpp
+++ b/include/ck_tile/ops/unified_attention/block/block_masking.hpp
@@ -180,6 +180,12 @@ struct GenericAttentionMask
         }
     }
 
+    // Alias kept for compatibility with reference_batched_masking.hpp
+    CK_TILE_HOST_DEVICE constexpr auto IsOutOfSinkBound(index_t i_y, index_t i_x) const
+    {
+        return IsOutOfBound(i_y, i_x);
+    }
+
     // per-pixel check if out-of-bound, if true, need mask a value(like -INF)
     CK_TILE_HOST_DEVICE constexpr auto IsOutOfBound(index_t i_y, index_t i_x) const
     {