博客详情

主页 / 博客详情
blog

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