v0.5.3 补丁更新
这是一个主要为了支持 CUDA 11.0 的补丁版本。现在 DGL 在 Linux/Windows/Mac 上支持 CUDA 11.0 和 PyTorch 1.7。
其他修复包括
- 图批量处理的性能修复。影响
dgl.batch
API。 - 图 readout 的加速。影响 所有 API under
dgl.readout
。 - 使用 sum reducer 的 CPU SpMM 加速。
- 移除 CPU 和 GPU 之间冗余拷贝的性能优化。
- 修复
segment_reduce()
忽略尾部 0 片段的问题。 - 修复由于未找到属性导致的崩溃错误。
- COO-CSR 转换中的性能优化。
- 异构图格式转换的并行化。
- 修复一个 bug,以启用使用 CPU 进行 RGCN 的分布式训练。
- 大量文档修复。
新示例
- 用于大型图的 GATNE-T 的稀疏嵌入。
- LINE。
- 适用于 OGB 的 SIGN。
中文用户指南已经发布了第 1 到第 4 章。后续章节将很快发布。
完整发布说明:https://github.com/dmlc/dgl/releases/tag/0.5.3
04 十二月