fix(build): correct preprocessor guard for GDN decode to fix Ampere c… (#52743)
Signed-off-by: prakharP <prakharpant288@gmail.com>
prakharPant committed
Aug 28, 2026 at 05:51 UTC
b2a6e9dca0c2e62142b96adafceb99fbeb3e60c3
1 file changed
+3
csrc/libtorch_stable/ops.h
+3
@@ -393,6 +393,9 @@ void fused_kda_decode(
393
std::optional<torch::stable::Tensor> output_gate,
394
std::optional<torch::stable::Tensor> norm_weight, double norm_eps);
395
396
+#endif
397
+
398
+#ifdef VLLM_ENABLE_FUSED_GDN_DECODE
399
void fused_gdn_decode_post_conv_mtp(
400
torch::stable::Tensor const& mixed_qkv, torch::stable::Tensor const& a,
401
torch::stable::Tensor const& b, torch::stable::Tensor const& a_log,