From 83dcee95b0d237d54f4a9d5b0b6440ec66c560cb Mon Sep 17 00:00:00 2001
From: dzzz2001 <dzcpku2001@163.com>
Date: Tue, 7 Apr 2026 14:07:46 +0800
Subject: [PATCH] fix compilation warning

---
 source/source_lcao/module_gint/kernel/gemm_nn_vbatch.cuh | 2 +-
 source/source_lcao/module_gint/kernel/gemm_tn_vbatch.cuh | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/source/source_lcao/module_gint/kernel/gemm_nn_vbatch.cuh b/source/source_lcao/module_gint/kernel/gemm_nn_vbatch.cuh
index 02c92b134bc..cac7abfd346 100644
--- a/source/source_lcao/module_gint/kernel/gemm_nn_vbatch.cuh
+++ b/source/source_lcao/module_gint/kernel/gemm_nn_vbatch.cuh
@@ -273,7 +273,7 @@ static __global__ void vbatched_gemm_nn_kernel(const int* M,
                                               const int* global_ldc,
                                               const T* alpha)
 {
-    extern __shared__ __align__(sizeof(T)) unsigned char smem[];
+    extern __shared__ __align__(sizeof(double)) unsigned char smem[];
     T* shared_mem = reinterpret_cast<T*>(smem);
 
     int batchid = blockIdx.z;
diff --git a/source/source_lcao/module_gint/kernel/gemm_tn_vbatch.cuh b/source/source_lcao/module_gint/kernel/gemm_tn_vbatch.cuh
index 8e56dc083e0..380eb9efa84 100644
--- a/source/source_lcao/module_gint/kernel/gemm_tn_vbatch.cuh
+++ b/source/source_lcao/module_gint/kernel/gemm_tn_vbatch.cuh
@@ -273,7 +273,7 @@ static __global__ void vbatched_gemm_nt_kernel(const int* M,
                                               const int* global_ldc,
                                               const T* alpha)
 {
-    extern __shared__ __align__(sizeof(T)) unsigned char smem[];
+    extern __shared__ __align__(sizeof(double)) unsigned char smem[];
     T* shared_mem = reinterpret_cast<T*>(smem);
 
     int batchid = blockIdx.z;