diff --git a/projects/clr/hipamd/src/hip_graph_internal.cpp b/projects/clr/hipamd/src/hip_graph_internal.cpp
index 4839b18af3..20f60a4c3f 100644
--- a/projects/clr/hipamd/src/hip_graph_internal.cpp
+++ b/projects/clr/hipamd/src/hip_graph_internal.cpp
@@ -271,8 +271,10 @@ bool Graph::TopologicalOrder(std::vector<Node>& TopoOrder) {
 // ================================================================================================
 void Graph::clone(Graph* newGraph, bool cloneNodes) const {
   newGraph->pOriginalGraph_ = this;
+  auto curDevId = ihipGetDevice();
   for (hip::GraphNode* entry : vertices_) {
     GraphNode* node = entry->clone();
+    node->SetDeviceId(curDevId);
     node->SetParentGraph(newGraph);
     newGraph->vertices_.push_back(node);
     newGraph->clonedNodes_[entry] = node;