mirrors / dmlc--dgl

项目文件夹

dmlc--dgl
文件
0.8.0post2
T
ayasar70 96cd2ee653 [Performance][GPU] Improve _SegmentCopyKernel() (#3470) 
* Based on issue #3436. Improving _SegmentCopyKernel s GPU utilization by switching to nonzero based thread assignment

* fixing lint issues

* Update cub for cuda 11.5 compatibility (#3468)

* fixing type mismatch

* tx guaranteed to be smaller than nnz. Hence removing last check

* minor: updating comment

* adding three unit tests for csr slice method to cover some corner cases

Co-authored-by: Abdurrahman Yasar <ayasar@nvidia.com>
Co-authored-by: nv-dlasalle <63612878+nv-dlasalle@users.noreply.github.com>
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2021-11-06 10:53:53 -07:00

3.3 KiB
原始文件 永久链接 Blame 文件历史

Contributing to DGL

Contribution is always welcomed. A good starting place is the roadmap issue, where you can find our current milestones. All contributions must go through pull requests and be reviewed by the committers. See our contribution guide for more details.

Once your contribution is accepted and merged, congratulations, you are now a contributor to the DGL project. We will put your name in the list below.

Contributors

在新工单中引用 查看 Git Blame 复制永久链接