SWDEV-540629 - Update the current device ID (#699)
[ROCm/clr commit: 0313e20706]
This commit is contained in:
1 parent
680c7fbf64
commit
52363f797e
2 files changed
+4
-2
No files matched your search
@@ -271,10 +271,8 @@ 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;
|
||||
|
||||
Reference in new issue
Block a user