From af64087732c2ff5253ce3eafb8c1aa651b548e75 Mon Sep 17 00:00:00 2001 From: Trunrain <270250579@qq.com> Date: Tue, 16 Dec 2025 17:36:40 +0800 Subject: [PATCH] [bugfix] matmul_allreduce_add_rmsnorm aclnn interface (#5082) What this PR does / why we need it? a2 kernel aclnn interface extern "C" fix Does this PR introduce any user-facing change? No How was this patch tested? vLLM version: v0.12.0 Signed-off-by: tongrunze Co-authored-by: tongrunze --- .../op_host/aclnn_matmul_allreduce_add_rmsnorm.cpp | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/csrc/matmul_allreduce_add_rmsnorm/op_host/aclnn_matmul_allreduce_add_rmsnorm.cpp b/csrc/matmul_allreduce_add_rmsnorm/op_host/aclnn_matmul_allreduce_add_rmsnorm.cpp index ec71fa91..396da512 100644 --- a/csrc/matmul_allreduce_add_rmsnorm/op_host/aclnn_matmul_allreduce_add_rmsnorm.cpp +++ b/csrc/matmul_allreduce_add_rmsnorm/op_host/aclnn_matmul_allreduce_add_rmsnorm.cpp @@ -26,6 +26,10 @@ enum NnopbaseHcclServerType { }; extern "C" void __attribute__((weak)) NnopbaseSetHcclServerType(void *executor, NnopbaseHcclServerType sType); +#ifdef __cplusplus +extern "C" { +#endif + extern aclnnStatus aclnnInnerMatmulAllreduceAddRmsnormGetWorkspaceSize( const aclTensor *x1, const aclTensor *x2, @@ -48,10 +52,6 @@ extern aclnnStatus aclnnInnerMatmulAllreduceAddRmsnorm( aclOpExecutor *executor, aclrtStream stream); -#ifdef __cplusplus -extern "C" { -#endif - aclnnStatus aclnnMatmulAllreduceAddRmsnormGetWorkspaceSize( const aclTensor *x1, const aclTensor *x2,