From e033b414c4fa2d76f1871c96d97594b14ba19e23 Mon Sep 17 00:00:00 2001
From: Varun Sundar Rabindranath <varun@neuralmagic.com>
Date: Wed, 18 Dec 2024 16:09:31 +0000
Subject: [PATCH] fix common header

Signed-off-by: Varun Sundar Rabindranath <varun@neuralmagic.com>
---
 csrc/quantization/cutlass_w8a8/scaled_mm_c3x.cuh | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/csrc/quantization/cutlass_w8a8/scaled_mm_c3x.cuh b/csrc/quantization/cutlass_w8a8/scaled_mm_c3x.cuh
index c5f8193537001..d4bc2f0ade50d 100644
--- 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
\ No newline at end of file
+}  // namespace vllm