o Ti@s2dZddlZddlZGdddeZdddZdS) zC Functionality of swapping tensors to/from (NVMe) storage devices. Nc@sFeZdZdddZddZddZdd Zd d Zd d ZddZ dS)JobNcCs||_||_||_d|_dSN)cmd_line output_filework_dir output_fd)selfrrrr M/home/ubuntu/.local/lib/python3.10/site-packages/deepspeed/nvme/ds_aio_job.py__init__s z Job.__init__cC|jSr)rrr r r cmdzJob.cmdcCr rrr r r r get_stdoutrzJob.get_stdoutcCr rrr r r r get_stderrrzJob.get_stderrcCr r)rr r r r get_cwdrz Job.get_cwdcCs |jdurt|jd|_dSdS)Nw)ropenrr r r r open_output_file s zJob.open_output_filecCs"|jdur|jd|_dSdSr)rcloser r r r close_output_file$s   zJob.close_output_file)NN) __name__ __module__ __qualname__r rrrrrrr r r r r s  rFcCsttd|}|rtd||tj||| | d}| |j dks8Jd|dS)N zargs = )argsstdoutstderrcwdrzThis command failed: ) shlexsplitjoinrprintr subprocessrunrrrr returncode)jobverboserprocr r r run_job*s  r+)F)__doc__r%r!objectrr+r r r r s