[Fix]atomicAdd ambiguous

f850f22a · zhuwenwen · 07e3681a · f850f22a
Commit f850f22a authored Mar 19, 2025 by zhuwenwen
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

csrc/quantization/gptq/q_gemm.cu csrc/quantization/gptq/q_gemm.cu +4 -0

No files found.
--- a/csrc/quantization/gptq/q_gemm.cu
+++ b/csrc/quantization/gptq/q_gemm.cu
@@ -19,6 +19,10 @@ https://github.com/qwopqwop200/GPTQ-for-LLaMa
 #include "qdq_4.cuh"
 #include "qdq_8.cuh"

+#ifndef USE_ROCM
+#include "compat.cuh"
+#endif
+
 namespace vllm {
 namespace gptq {