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

SWDEV-408701 - Avoid adding duplicates when no node is captured in cross stream

Browse Source 
Change-Id: Iccbddaf9db73238aa938f8b3e8ec410f4a86d524
This commit is contained in:
Anusha GodavarthySurya
2023-10-09 06:42:08 +00:00
committed by Anusha Godavarthy Surya
parent 2535fac296
commit c7e4e9c2f4
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
hipamd/src/hip_internal.hpp
+4 -1
View File
@@ -336,7 +336,10 @@ namespace hip {
lastCapturedNodes_.clear(); lastCapturedNodes_.clear();
} }
for (auto node : graphNodes) { for (auto node : graphNodes) {
lastCapturedNodes_.push_back(node); if (std::find(lastCapturedNodes_.begin(), lastCapturedNodes_.end(), node) ==
lastCapturedNodes_.end()) {
lastCapturedNodes_.push_back(node);
}
} }
} }
/// Set graph that is being captured /// Set graph that is being captured