Skip to content

Commit

Permalink
More wrappers.
Browse files Browse the repository at this point in the history
  • Loading branch information
harrism committed Sep 4, 2024
1 parent ad011e0 commit d9fae9f
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
6 changes: 3 additions & 3 deletions cpp/include/cugraph/legacy/graph.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -354,7 +354,7 @@ class GraphCOO {
edge_t number_of_edges,
bool has_data = false,
cudaStream_t stream = nullptr,
rmm::device_async_resource_ref mr = raft::resource::get_current_device_resource_ref()())
rmm::device_async_resource_ref mr = raft::resource::get_current_device_resource())
: number_of_vertices_p(number_of_vertices),
number_of_edges_p(number_of_edges),
src_indices_p(sizeof(vertex_t) * number_of_edges, stream, mr),
Expand All @@ -365,7 +365,7 @@ class GraphCOO {

GraphCOO(GraphCOOView<vertex_t, edge_t, weight_t> const& graph,
cudaStream_t stream = nullptr,
rmm::device_async_resource_ref mr = raft::resource::get_current_device_resource_ref()())
rmm::device_async_resource_ref mr = raft::resource::get_current_device_resource())
: number_of_vertices_p(graph.number_of_vertices),
number_of_edges_p(graph.number_of_edges),
src_indices_p(graph.src_indices, graph.number_of_edges * sizeof(vertex_t), stream, mr),
Expand Down Expand Up @@ -530,7 +530,7 @@ class GraphCSR : public GraphCompressedSparseBase<vertex_t, edge_t, weight_t> {
edge_t number_of_edges_,
bool has_data_ = false,
cudaStream_t stream = nullptr,
rmm::device_async_resource_ref mr = raft::resource::get_current_device_resource_ref()())
rmm::device_async_resource_ref mr = raft::resource::get_current_device_resource())
: GraphCompressedSparseBase<vertex_t, edge_t, weight_t>(
number_of_vertices_, number_of_edges_, has_data_, stream, mr)
{
Expand Down
2 changes: 1 addition & 1 deletion cpp/tests/utilities/base_fixture.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -54,7 +54,7 @@ namespace test {
* ```
**/
class BaseFixture : public ::testing::Test {
rmm::device_async_resource_ref _mr{rmm::mr::get_current_device_resource_ref()};
rmm::device_async_resource_ref _mr{raft::resource::get_current_device_resource_ref()};

public:
/**
Expand Down

0 comments on commit d9fae9f

Please sign in to comment.