github-featured / dmlc--dgl

发布

0 发布 42 Git 标签

RSS 订阅源
  • 0.9.1 960092be02
    比较

    [Feature] Import PyTorch's CUDA stream management (#4503)

    frostbyte_neo 发布于 2022-09-16 02:50:27 +00:00 | 1964 次提交 在此版本后已推送到 master

    • add set_stream

    • add .record_stream for NDArray and HeteroGraph

    • refactor dgl stream Python APIs

    • test record_stream

    • add unit test for record stream

    • use pytorch's stream

    • fix lint

    • fix cpu build

    • address comments

    • address comments

    • add record stream tests for dgl.graph

    • record frames and update dataloder

    • add docstring

    • update frame

    • add backend check for record_stream

    • remove CUDAThreadEntry::stream

    • record stream for newly created formats

    • fix bug

    • fix cpp test

    • fix None c_void_p to c_handle

    下载附件