We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 45ce85e commit 677f67eCopy full SHA for 677f67e
1 file changed
cpp/src/neighbors/detail/cagra/cagra_helpers.cpp
@@ -86,6 +86,8 @@ std::tuple<size_t, size_t, size_t, size_t> optimize_workspace_size(size_t n_rows
86
87
size_t debug_host_size = 0;
88
if (raft::default_logger().should_log(rapids_logger::level_enum::debug)) {
89
+ // cagra::detail::graph::optimize() allocates extra memory to calculate
90
+ // graph metrics when debug logging is enabled
91
debug_host_size = n_rows * graph_degree * sizeof(uint32_t) // host_copy_output_graph
92
+ n_rows * sizeof(uint32_t) // in_edge_count
93
+ graph_degree * sizeof(uint32_t); // hist
0 commit comments