github-featured / deepspeedai--deepspeed

发布

0 发布 128 Git 标签

RSS 订阅源
  • v0.15.3 a24cdd6b67
    比较

    [XPU] [DeepNVMe] use same cpu_op_desc_t with cuda (#6645)

    frostbyte_neo 发布于 2024-10-22 14:45:05 +00:00

    We have found that #6592 uses _pinned_tensor_mgr to create cpu bounce
    buffer, which is same with what our xpu accelerator currently doing.
    So no need to use xpu device specific cpu_op_desc_t.
    In this PR:

    1. remove custom csrc/xpu/aio/deepspeed_cpu_op.cpp
    2. modify xpu async_io opbuilder.

    This issue cannot be easily done with revert #6532 , for we added some
    source file as last time GDS feature going in DS. So file this new PR :)

    下载附件