[DistGB] enable non-int64_t index tensor
- Dominant language
- Python
- Stars
- 14.3k
- Forks
- 3.1k
- PR merge metrics
- No merged PRs in 30d
Description
## 🔨Work Item
**IMPORTANT:**
* This template is only for dev team to track project progress. For feature request or bug report, please use the corresponding issue templates.
* DO NOT create a new work item if the purpose is to fix an existing issue or feature request. We will directly use the issue in the project tracker.
Project tracker: https://github.com/orgs/dmlc/projects/2
## Description
For now, we always expect int64_t in index tensor while torch supports non-int64_t index tensor since `2.0.0`. Such limitation hinders we use NID/EID with int32_t. And such index tensor is often used as accessing `DistTensor`.
https://github.com/dmlc/dgl/blob/73e01d6daadaba537053a75474237286456e8f12/src/rpc/rpc.cc#L366-L370
## Depending work items or issues
Contributor guide
No contributing guide indexed for this repository
Assessment
This issue has not been assessed yet.