Putting back header inclusion (seems unused but still)

2023-07-20 15:46:51 +00:00 · 2023-07-20 15:46:51 +00:00 · f555dabca8
parent 5ca0508d02
commit f555dabca8
1 changed files with 1 additions and 1 deletions
--- 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>