From e4794a2ab0048c478f71b503a11efa04806040a6 Mon Sep 17 00:00:00 2001 From: mengyuanli Date: Tue, 5 Aug 2025 11:24:16 +0800 Subject: [PATCH] compile bug --- ccsrc/ops/ascendc/add_rms_norm.cc | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/ccsrc/ops/ascendc/add_rms_norm.cc b/ccsrc/ops/ascendc/add_rms_norm.cc index a9f47a2..0a20f9f 100644 --- a/ccsrc/ops/ascendc/add_rms_norm.cc +++ b/ccsrc/ops/ascendc/add_rms_norm.cc @@ -118,7 +118,8 @@ private: }; } // namespace ms_custom_ops -MS_CUSTOM_OPS_REGISTER(add_rms_norm, AddRmsNormCustomOpFuncImpl, AddRmsNormCustomAscend); +REG_GRAPH_MODE_OP(add_rms_norm, ms_custom_ops::AddRmsNormCustomOpFuncImpl, + ms_custom_ops::AddRmsNormCustomAscend); // ============================================================================= // PYBOOST MODE IMPLEMENTATION -- Gitee