From 8bfb7ed2f246dbd63c8fd15d2afe8616172e2b1f Mon Sep 17 00:00:00 2001
From: bssrdf <bssrdf@gmail.com>
Date: Thu, 13 Nov 2025 16:32:27 -0500
Subject: [PATCH] restore smem pointer at teh end of evry rs loop

---
 ggml/src/ggml-cuda/conv2d-implicit.cu | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/ggml/src/ggml-cuda/conv2d-implicit.cu b/ggml/src/ggml-cuda/conv2d-implicit.cu
index bd67ac2b86..d2b5ee6d33 100644
--- a/ggml/src/ggml-cuda/conv2d-implicit.cu
+++ b/ggml/src/ggml-cuda/conv2d-implicit.cu
@@ -949,6 +949,8 @@ static __global__ void conv2d_implicit_kernel(const half * __restrict__ input,
       s = 0;
       r++;
     }
+    A_block_smem = shmem;
+    B_block_smem = &shmem[BM * BK];
   } // iter r
 
     // reuse smem