Skip to content

Commit 9e28c23

Browse files
drisspgYuqi Zhang
authored andcommitted
Fix Broken macro for cutlass moe (vllm-project#18049)
Signed-off-by: drisspg <[email protected]> Signed-off-by: Yuqi Zhang <[email protected]>
1 parent a6d9c31 commit 9e28c23

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

csrc/quantization/cutlass_w8a8/scaled_mm_entry.cu

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,8 @@ void cutlass_scaled_mm_sm90(torch::Tensor& c, torch::Tensor const& a,
2929
torch::Tensor const& a_scales,
3030
torch::Tensor const& b_scales,
3131
std::optional<torch::Tensor> const& bias);
32-
32+
#endif
33+
#if defined ENABLE_CUTLASS_MOE_SM90 && ENABLE_CUTLASS_MOE_SM90
3334
void cutlass_moe_mm_sm90(
3435
torch::Tensor& out_tensors, torch::Tensor const& a_tensors,
3536
torch::Tensor const& b_tensors, torch::Tensor const& a_scales,

0 commit comments

Comments
 (0)