Skip to content
Merged
Show file tree
Hide file tree
Changes from 3 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions csrc/multidevice/ipc_handle.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -143,6 +143,9 @@ void IpcHandleCache::exchangeHandles(
insert(communication, std::move(ipc_handles));
}

if (non_cached_communications.empty()) {
return;
}
// a barrier is needed here to ensure all ranks have received the
// memhandles and the keys are deleted from the store before the next call to
// exchangeHandles, otherwise there is a correctness issue
Expand Down
4 changes: 3 additions & 1 deletion tests/cpp/test_multidevice_host_ir_overlap.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -1261,7 +1261,9 @@ TEST_F(
hic->addOutput(tva_j_next_slice);
hic->addOutput(tvc_j);

hir::HostIrEvaluator hie(std::move(hic), communicator_);
HostIrEvaluatorParams params;
params.use_allocation_cache = true;
hir::HostIrEvaluator hie(std::move(hic), communicator_, params);

at::manual_seed(getATenRandomSeed());

Expand Down
Loading