From 4906961bf1fd22a5e44b27f926bf3b12775ef384 Mon Sep 17 00:00:00 2001
From: John Smith <yfshi123@163.com>
Date: Thu, 23 Mar 2023 23:37:39 +0800
Subject: [PATCH] fix bug

---
 GPTQ-for-LLaMa/quant_cuda_kernel.cu | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/GPTQ-for-LLaMa/quant_cuda_kernel.cu b/GPTQ-for-LLaMa/quant_cuda_kernel.cu
index de0c0d6..ed9627a 100644
--- a/GPTQ-for-LLaMa/quant_cuda_kernel.cu
+++ b/GPTQ-for-LLaMa/quant_cuda_kernel.cu
@@ -631,7 +631,7 @@ __global__ void VecQuant4TransposeMatMulHalfKernel(
 #if __CUDA_ARCH__ < 700 && __CUDA_ARCH__ > 600
   atomicAddHalf(&mul2[n_cols * height * 8 + n_rows], res);
 #else
-  atomicAddHalf(&mul2[n_cols * height * 8 + n_rows], res);
+  atomicAdd(&mul2[n_cols * height * 8 + n_rows], res);
 #endif
 #endif
 }