From fbb90ef3a76cd6c469782c84668d25c5f3bfdd22 Mon Sep 17 00:00:00 2001
From: Davit Grigoryan <ggc.language@gmail.com>
Date: Tue, 28 Apr 2026 22:06:08 +0000
Subject: critical fix - add option to have masks for SB entries

---
 src/abstract_hardware_model.cc | 8 ++++++++
 1 file changed, 8 insertions(+)

(limited to 'src/abstract_hardware_model.cc')

diff --git a/src/abstract_hardware_model.cc b/src/abstract_hardware_model.cc
index 72639d4..5d95db6 100644
--- a/src/abstract_hardware_model.cc
+++ b/src/abstract_hardware_model.cc
@@ -305,6 +305,14 @@ void warp_inst_t::set_split_id_on_sets(unsigned split_id) {
   }
 }
 
+void warp_inst_t::set_source_mask_on_sets(const active_mask_t &mask) {
+  for (unsigned s = 0; s < m_simd_sets.size(); s++) {
+    if (m_simd_sets[s].valid) {
+      m_simd_sets[s].source_mask = mask;
+    }
+  }
+}
+
 void warp_inst_t::merge_simd_sets(
     const std::vector<simd_set_info> &other_sets) {
   assert(m_simd_sets.size() == other_sets.size());
-- 
cgit v1.3