diff --git a/server/custom_kernels/custom_kernels/exllama/exllama_ext.cpp b/server/custom_kernels/custom_kernels/exllama/exllama_ext.cpp
index 4e43d605..b786988b 100644
--- a/server/custom_kernels/custom_kernels/exllama/exllama_ext.cpp
+++ b/server/custom_kernels/custom_kernels/exllama/exllama_ext.cpp
@@ -3,7 +3,7 @@
 #include <torch/extension.h>
 #include <c10/cuda/CUDAGuard.h>
 #include <ATen/cuda/CUDAContext.h>
-// #include <cuda_runtime.h>
+#include <cuda_runtime.h>
 #include <cuda_fp16.h>
 #include <cstdint>
 #include <cstdio>