diff --git a/ccsrc/ops/ascendc/add_rms_norm.cc b/ccsrc/ops/ascendc/add_rms_norm.cc
index a9f47a260b9e89de49dc51d88587a19a49e4e607..0a20f9fe0027e44b5863428f9d25a5d19e180dd6 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