diff --git a/csrc/quantization/cutlass_w8a8/scaled_mm_c3x.cuh b/csrc/quantization/cutlass_w8a8/scaled_mm_c3x.cuh index c5f8193537001..83dedead6fa34 100644 --- a/csrc/quantization/cutlass_w8a8/scaled_mm_c3x.cuh +++ b/csrc/quantization/cutlass_w8a8/scaled_mm_c3x.cuh @@ -17,7 +17,7 @@ #include "cutlass/epilogue/collective/collective_builder.hpp" #include "cutlass/gemm/collective/collective_builder.hpp" -#include "common.hpp" +#include "cutlass_extensions/common.hpp" // clang-format on /* @@ -156,4 +156,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