[Kernel] Update cutlass_scaled_mm to support 2d group (blockwise) scaling (#11868)

2025-01-30 21:33:00 -05:00
parent 4078052f09
commit 9798b2fb00
25 changed files with 1924 additions and 346 deletions
--- a/csrc/cutlass_extensions/vllm_collective_builder.cuh
+++ b/csrc/cutlass_extensions/vllm_collective_builder.cuh
@@ -1,6 +1,6 @@
 #pragma once

-#include "cutlass/gemm/collective/collective_builder.hpp"
+#include "cutlass_extensions/gemm/collective/collective_builder.hpp"

 namespace cutlass::gemm::collective {
 using namespace cute;