Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Migrate from raft::device_resources -> raft::resources #1510

Merged
merged 10 commits into from
May 16, 2023
Merged

Migrate from raft::device_resources -> raft::resources #1510

Changes from 1 commit
Commits
Show all changes
10 commits
Select commit Hold shift + click to select a range
c8eb634
Migrate from raft::device_resources -> raft::resources
benfred May 10, 2023
e624c38
fix merge
benfred May 11, 2023
a85dc76
initialize cuda_stream_pool factory
benfred May 11, 2023
b026ca7
add missing includes
benfred May 14, 2023
b78a40c
Merge branch 'branch-23.06' into resources
benfred May 15, 2023
d69b6e5
re-apply resource to recent changes
benfred May 15, 2023
45d596c
fix mpi_comms
benfred May 15, 2023
cad2a3e
fix
benfred May 16, 2023
88133b2
Merge branch 'branch-23.06' into resources
benfred May 16, 2023
4cd8815
apply suggestions from code review
benfred May 16, 2023
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
Prev Previous commit
Next Next commit
fix mpi_comms
  • Loading branch information
@benfred
benfred committed May 15, 2023
commit 45d596c40945c75ba7220b6feb71fd8ca85db1e7
2 changes: 1 addition & 1 deletion cpp/include/raft/comms/mpi_comms.hpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -60,7 +60,7 @@ using mpi_comms = detail::mpi_comms;
inline void initialize_mpi_comms(resources* handle, MPI_Comm comm)
{
auto communicator = std::make_shared<comms_t>(
std::unique_ptr<comms_iface>(new mpi_comms(comm, false, handle->get_stream())));
std::unique_ptr<comms_iface>(new mpi_comms(comm, false, resource::get_cuda_stream(*handle))));
resource::set_comms(*handle, communicator)
};

Expand Down