Fix compile error when using rocm (#2648)

2024-02-02 01:35:09 +08:00
parent cd9e60c76c
commit 923797fea4
3 changed files with 9 additions and 1 deletions
--- a/csrc/quantization/fp8_e5m2_kvcache/quant_utils.cuh
+++ b/csrc/quantization/fp8_e5m2_kvcache/quant_utils.cuh
@@ -9,7 +9,6 @@
 #include "../../attention/dtype_float16.cuh"
 #include "../../attention/dtype_bfloat16.cuh"

-#pragma once

 namespace vllm {
 #ifdef ENABLE_FP8_E5M2