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,