diff --git a/csrc/libtorch_stable/ops.h b/csrc/libtorch_stable/ops.h index b0c8d387ba..31983fc372 100644 --- a/csrc/libtorch_stable/ops.h +++ b/csrc/libtorch_stable/ops.h @@ -393,6 +393,9 @@ void fused_kda_decode( std::optional output_gate, std::optional norm_weight, double norm_eps); +#endif + +#ifdef VLLM_ENABLE_FUSED_GDN_DECODE void fused_gdn_decode_post_conv_mtp( torch::stable::Tensor const& mixed_qkv, torch::stable::Tensor const& a, torch::stable::Tensor const& b, torch::stable::Tensor const& a_log,