dmlc / dmlc/dgl

Replace custom cuda hashtable with cuCollections'

Open
#5,498 1 comment 3 reactions 1 assignee Claimed by @TristonC View on GitHub
feature request topic: system performance
Dominant language
Python
Stars
14.3k
Forks
3.1k
PR merge metrics
No merged PRs in 30d

Description

## 🚀 Feature
Currently, the most expensive part of GPU based sampling is running to_block(), and specifically hash table insertions. The current implementation [cuda_hashtable.cuh](https://github.com/dmlc/dgl/blob/master/src/runtime/cuda/cuda_hashtable.cuh) does not make good use of the hardware, and instead we should replace it the implementation in [cuCollections](https://github.com/NVIDIA/cuCollections/blob/dev/include/cuco/static_map.cuh#L44).

See https://developer.nvidia.com/blog/maximizing-performance-with-massively-parallel-hash-maps-on-gpus/ for a more in-depth explanation.

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.