diff --git a/ge/hybrid/model/hybrid_model_builder.cc b/ge/hybrid/model/hybrid_model_builder.cc
index 8fa985c1..8ca66a71 100755
--- a/ge/hybrid/model/hybrid_model_builder.cc
+++ b/ge/hybrid/model/hybrid_model_builder.cc
@@ -127,7 +127,7 @@ Status CollectDependenciesForFusedGraph(NodeItem &node_item, std::set<OpDesc *>
   return SUCCESS;
 }
 
-bool CheckHasHostMem(NodeItem &src_node_item) {
+bool CheckHasHostMem(NodeItem *src_node_item) {
   if (src_node_item == nullptr) {
     return false;
   }