Skip to content

GitLab

Unverified Commit e57e4d6e authored by Driss Guessous's avatar Driss Guessous Committed by GitHub
Browse files

Fix Broken macro for cutlass moe (#18049) 


Signed-off-by: default avatardrisspg <drisspguessous@gmail.com>
parent ee5be834
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
csrc/quantization/cutlass_w8a8/scaled_mm_entry.cu
View file @ e57e4d6e
...@@ -29,7 +29,8 @@ void cutlass_scaled_mm_sm90(torch::Tensor& c, torch::Tensor const& a, ...@@ -29,7 +29,8 @@ void cutlass_scaled_mm_sm90(torch::Tensor& c, torch::Tensor const& a,
torch::Tensor const& a_scales, torch::Tensor const& a_scales,
torch::Tensor const& b_scales, torch::Tensor const& b_scales,
std::optional<torch::Tensor> const& bias); std::optional<torch::Tensor> const& bias);
#endif
#if defined ENABLE_CUTLASS_MOE_SM90 && ENABLE_CUTLASS_MOE_SM90
void cutlass_moe_mm_sm90( void cutlass_moe_mm_sm90(
torch::Tensor& out_tensors, torch::Tensor const& a_tensors, torch::Tensor& out_tensors, torch::Tensor const& a_tensors,
torch::Tensor const& b_tensors, torch::Tensor const& a_scales, torch::Tensor const& b_tensors, torch::Tensor const& a_scales,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment