From 141bd913e1d148ec180f1162f7e2eaf32a74e954 Mon Sep 17 00:00:00 2001 From: Trunrain <270250579@qq.com> Date: Fri, 19 Dec 2025 17:06:59 +0800 Subject: [PATCH] restore matmul_allreduce_add_rmsnrom aclnn interface (#5119) **What this PR does / why we need it?** restore a2 matmul_allreduce_add_rmsnrom kernel aclnn interface **How was this patch tested?** - vLLM version: v0.12.0 - vLLM main: https://github.com/vllm-project/vllm/commit/ad32e3e19ccf0526cb6744a5fed09a138a5fb2f9 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,