Merge branch 'main' into rd/recip_fp16b_acc

tenstorrent · Aug 7, 2024 · e76df6c · e76df6c
2 parents 1a0243c + 3a492fb
commit e76df6c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/llk_lib/llk_math_common.h b/llk_lib/llk_math_common.h
@@ -87,6 +87,8 @@ inline void _llk_math_pack_sync_init_() {
 
 template <bool mail2math=true, bool mail2pack=true>
 inline void _llk_math_get_tile_(std::uint32_t tile_index, std::uint32_t* p_tile) {
+    constexpr uint32_t wait_sem = (mail2math && mail2pack) ? (2) : (1);
+    while (semaphore_read(semaphore::UNPACK_OPERAND_SYNC) < wait_sem);
     if constexpr (mail2math) {
        *p_tile = mailbox_read(ThreadId::UnpackThreadId);
     } else {

diff --git a/llk_lib/llk_pack_common.h b/llk_lib/llk_pack_common.h
@@ -160,6 +160,8 @@ inline void _llk_pack_dest_init_(const std::uint32_t face_r_dim = FACE_R_DIM, co
 
 template <bool mail2math=true, bool mail2pack=true>
 inline void _llk_pack_get_tile_(std::uint32_t tile_index, std::uint32_t *p_tile) {
+    constexpr uint32_t wait_sem = (mail2math && mail2pack) ? (2) : (1);
+    while (semaphore_read(semaphore::UNPACK_OPERAND_SYNC) < wait_sem);
     if constexpr (mail2pack) {
        *p_tile = mailbox_read(ThreadId::UnpackThreadId);
     } else {