fix common header

Signed-off-by: Varun Sundar Rabindranath <[email protected]>
vllm-project · Dec 19, 2024 · e033b41 · e033b41
1 parent 4068c8e
commit e033b41
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/csrc/quantization/cutlass_w8a8/scaled_mm_c3x.cuh b/csrc/quantization/cutlass_w8a8/scaled_mm_c3x.cuh
@@ -17,7 +17,8 @@
 #include "cutlass/epilogue/collective/collective_builder.hpp"
 #include "cutlass/gemm/collective/collective_builder.hpp"
 
-#include "common.hpp"
+#include "core/math.hpp"
+#include "cutlass_extensions/common.hpp"
 // clang-format on
 
 /*
@@ -156,4 +157,4 @@ void cutlass_gemm_caller(torch::Tensor& out, torch::Tensor const& a,
   CUTLASS_CHECK(status);
 }
 
-}  // namespace vllm
+}  // namespace vllm