File tree 1 file changed +4
-3
lines changed
1 file changed +4
-3
lines changed Original file line number Diff line number Diff line change @@ -535,10 +535,11 @@ create_graph_from_partitioned_edgelist(
535
535
536
536
auto total_global_mem = handle.get_device_properties ().totalGlobalMem ;
537
537
auto constexpr mem_frugal_ratio =
538
- 0.05 ; // if the expected temporary buffer size exceeds the mem_frugal_ratio of the
539
- // total_global_mem , switch to the memory frugal approach
538
+ 0.5 ; // if the aggregate edge data size exceeds the mem_frugal_ratio of the total_global_mem
539
+ // (in an approximate sense) , switch to the memory frugal approach
540
540
auto mem_frugal_threshold =
541
- static_cast <size_t >(static_cast <double >(total_global_mem / element_size) * mem_frugal_ratio);
541
+ static_cast <size_t >(static_cast <double >(total_global_mem / element_size) * mem_frugal_ratio) /
542
+ static_cast <size_t >(minor_comm_size);
542
543
543
544
std::vector<rmm::device_uvector<edge_t >> edge_partition_offsets;
544
545
std::vector<rmm::device_uvector<vertex_t >> edge_partition_indices;
You can’t perform that action at this time.
0 commit comments