balanced-k-means: fix a too large initial memory pool size#1148

Merged

rapids-bot[bot] merged 2 commits intorapidsai:branch-23.02from achirkin:bug-balanced-kmeans-bad-pool-size

Jan 19, 2023

+31-23

Commits on Jan 17, 2023

Limit the initial pool size
achirkin
committed

Commits on Jan 18, 2023

return mem_per_row for better handling of the workspace size
achirkin
committed