Skip to content

Commit 6717cca

Browse files
matthewdcongmatthewdcong
authored
Improve load balancing in DistributedPointsToGrid (#2276)
* Balance the multi-GPU merge tree across every device Signed-off-by: Matthew Cong <mcong@nvidia.com> * Generalize merge to non power-of-two GPU counts Signed-off-by: Matthew Cong <mcong@nvidia.com> * Remove prefetches causing performance regressions Signed-off-by: Matthew Cong <mcong@nvidia.com> --------- Signed-off-by: Matthew Cong <mcong@nvidia.com>
1 parent e538a06 commit 6717cca

2 files changed

Lines changed: 176 additions & 200 deletions

File tree

0 commit comments

Comments
 (0)