diff --git a/hipamd/src/hip_graph.cpp b/hipamd/src/hip_graph.cpp index eb31803937..754d3ecb01 100644 --- a/hipamd/src/hip_graph.cpp +++ b/hipamd/src/hip_graph.cpp @@ -1295,6 +1295,7 @@ hipError_t ihipGraphInstantiate(hip::GraphExec** pGraphExec, hip::Graph* graph, } std::unordered_map clonedNodes; hip::Graph* clonedGraph = graph->clone(clonedNodes); + clonedGraph->memAllocNodePtrs_ = graph->memAllocNodePtrs_; if (clonedGraph == nullptr) { return hipErrorInvalidValue; }