Avoid multiple redefinition (vllm-project#1817)

jikunshang · Dec 14, 2023 · 614856d · 614856d
1 parent 05bdf4e
commit 614856d
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 0 deletions.
diff --git a/csrc/cache.h b/csrc/cache.h
@@ -1,3 +1,5 @@
+#pragma once
+
 #include <torch/extension.h>
 
 #include <map>

diff --git a/csrc/cuda_utils.h b/csrc/cuda_utils.h
@@ -1,3 +1,5 @@
+#pragma once
+
 #include <torch/extension.h>
 
 int get_device_attribute(

diff --git a/csrc/dispatch_utils.h b/csrc/dispatch_utils.h
@@ -2,6 +2,8 @@
  * Adapted from
  * https://github.com/pytorch/pytorch/blob/v2.0.1/aten/src/ATen/Dispatch.h
  */
+#pragma once
+
 #include <torch/extension.h>
 
 #define VLLM_DISPATCH_CASE_FLOATING_TYPES(...)              \

diff --git a/csrc/ops.h b/csrc/ops.h
@@ -1,3 +1,5 @@
+#pragma once
+
 #include <torch/extension.h>
 
 void paged_attention_v1(