Skip to content

Commit

Permalink
compilation error
Browse files Browse the repository at this point in the history
  • Loading branch information
tarang-jain committed May 14, 2024
1 parent 895ea1f commit 2fca377
Showing 1 changed file with 6 additions and 6 deletions.
12 changes: 6 additions & 6 deletions faiss/gpu/StandardGpuResources.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -260,9 +260,9 @@ void StandardGpuResourcesImpl::setDefaultStream(
#if defined USE_NVIDIA_RAFT
// delete the raft handle for this device, which will be initialized
// with the updated stream during any subsequent calls to getRaftHandle
auto it = raftHandles_.find(device);
if (it != raftHandles_.end()) {
raftHandles_.erase(it);
auto it2 = raftHandles_.find(device);
if (it2 != raftHandles_.end()) {
raftHandles_.erase(it2);
}
#endif
}
Expand All @@ -286,9 +286,9 @@ void StandardGpuResourcesImpl::revertDefaultStream(int device) {
#if defined USE_NVIDIA_RAFT
// delete the raft handle for this device, which will be initialized
// with the updated stream during any subsequent calls to getRaftHandle
auto it = raftHandles_.find(device);
if (it != raftHandles_.end()) {
raftHandles_.erase(it);
auto it2 = raftHandles_.find(device);
if (it2 != raftHandles_.end()) {
raftHandles_.erase(it2);
}
#endif
}
Expand Down

0 comments on commit 2fca377

Please sign in to comment.