o Ti @s0ddlZddlZddlmZGdddeZdS)N) CPUOpBuildercsneZdZdZdZfddZddZddZd d Zd d Z fd dZ ddZ ddZ dfdd Z ZS)AsyncIOBuilder DS_BUILD_AIOasync_iocstj|jddS)N)name)super__init__NAMEself __class__Y/home/ubuntu/.local/lib/python3.10/site-packages/deepspeed/ops/op_builder/cpu/async_io.pyr szAsyncIOBuilder.__init__cCsd|jdS)Nzdeepspeed.ops.aio._op)r r rrr absolute_namezAsyncIOBuilder.absolute_namecCs gd}|S)N) z*csrc/aio/py_lib/deepspeed_py_io_handle.cppz$csrc/aio/py_lib/deepspeed_py_aio.cppz+csrc/aio/py_lib/deepspeed_py_aio_handle.cppz(csrc/aio/py_lib/deepspeed_aio_thread.cppz'csrc/aio/common/deepspeed_aio_utils.cppz(csrc/aio/common/deepspeed_aio_common.cppz'csrc/aio/common/deepspeed_aio_types.cppz$csrc/aio/py_lib/deepspeed_cpu_op.cppz)csrc/aio/py_lib/deepspeed_aio_op_desc.cppz%csrc/aio/py_lib/deepspeed_py_copy.cppz(csrc/aio/py_lib/deepspeed_pin_tensor.cppr)r src_listrrr lib_sourcesszAsyncIOBuilder.lib_sourcescCs|dgS)Nzcsrc/aio/py_lib/py_ds_aio.cpp)rr rrrsources!rzAsyncIOBuilder.sourcescCddgS)Nzcsrc/aio/py_libzcsrc/aio/commonrr rrr include_paths$zAsyncIOBuilder.include_pathscsdt}ddl}tt|jddd\}}|dkr |dks*|d|d|gd7}|S)Nr.rz -std=c++17z -std=c++14)z-Wallz-O0z-sharedz-fPICz -Wno-reorder) rcxx_argstorchmapint __version__splitremoveappend)r argsr TORCH_MAJOR TORCH_MINORr rrr's    zAsyncIOBuilder.cxx_argscCr)Nz-laioz-fopenmprr rrr extra_ldflags2rzAsyncIOBuilder.extra_ldflagsc Cstgdgdgdd}d}|D]=\}}|\}}}t|}|durP|||g} tj| tjtjd} | dkr?d}|S||j d |d ||Sq|S) N)z-lz libaio-devapt)z-Qlibaiopacman)z-qz libaio-develyum)dpkgr*rpmF)stdoutstderrrTz: please install the z package with ) dictitemsshutilwhich subprocessPopenPIPEwaitwarningr ) r libsfoundpkgmgrdataflaglibtoolpathcmdresultrrrcheck_for_libaio_pkg5s*    z#AsyncIOBuilder.check_for_libaio_pkgFcsH|dd}|r|s||jd||dt|o#|S)N io_submit)aiozI requires the dev libaio .so object and headers but these were not found.zIf libaio is already installed (perhaps from source), try setting the CFLAGS and LDFLAGS environment variables to where it can be found.) has_functionr8r rCr is_compatible)r verboseaio_compatibler rrrGJs zAsyncIOBuilder.is_compatible)F)__name__ __module__ __qualname__ BUILD_VARr r rrrrrr'rCrG __classcell__rrr rr s    r)r2r4builderrrrrrrs