AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

SWDEV-530803 - User current device id while cloning graph node. (#313)

Browse Source 
[ROCm/clr commit: 32eb6a5d89]
This commit is contained in:
Patel, Jaydeepkumar
2025-05-14 20:36:15 -07:00
committed by GitHub
parent b13eec7049
commit 7b6e59cbea
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/clr/hipamd/src/hip_graph_internal.cpp
+2
View File
@@ -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;