DeepSpeed
a1f98bdc - AIO CPU Locked Tensor (#6592)
Commit
View On GitHub
Comment changes are shownComment changes are hidden
Commit
308 days ago
AIO CPU Locked Tensor (#6592) Restoring the functionality of the cpu locked tensor in the AIO library. Make async_io operator available for CPU accelerator, i.e., CPU only environment. --------- Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com>
Author
jomayeri jomayeri
Parents
7d751ee8
  • accelerator
    • File
      cpu_accelerator.py
  • csrc
    • aio
      • common
        • File
          deepspeed_aio_common.cpp
        • File
          deepspeed_aio_common.h
        • File
          deepspeed_aio_utils.cpp
        • File
          deepspeed_aio_utils.h
      • py_lib
        • File
          deepspeed_aio_op_desc.cpp
        • File
          deepspeed_aio_op_desc.h
        • File
          deepspeed_cpu_op.cpp
        • File
          deepspeed_cpu_op.h
        • File
          deepspeed_pin_tensor.cpp
        • File
          deepspeed_pin_tensor.h
        • File
          deepspeed_py_aio.cpp
        • File
          deepspeed_py_aio_handle.cpp
        • File
          deepspeed_py_aio_handle.h
        • File
          deepspeed_py_io_handle.cpp
        • File
          deepspeed_py_io_handle.h
        • File
          py_ds_aio.cpp
    • gds/py_lib
      • File
        deepspeed_gds_op.cpp
      • File
        deepspeed_gds_op.h
      • File
        deepspeed_py_gds_handle.cpp
      • File
        deepspeed_py_gds_handle.h
      • File
        py_ds_gds.cpp
  • docs/_tutorials
    • File
      deepnvme.md
  • op_builder
    • File
      builder.py
    • cpu
      • File
        __init__.py
      • File
        async_io.py
  • tests/unit
    • ops/aio
      • File
        test_aio.py
      • File
        test_gds.py
    • runtime/zero
      • File
        test_nvme_checkpointing.py
FAQ Terms Privacy Refunds Impressum