mirror of
https://github.com/deepspeedai/DeepSpeed.git
synced 2025-10-20 15:33:51 +08:00
a5400974df
Add performance tuning utilities: `ds_nvme_tune` and `ds_io`. Update tutorial with tuning section. --------- Co-authored-by: Ubuntu <jomayeri@microsoft.com> Co-authored-by: Joe Mayer <114769929+jomayeri@users.noreply.github.com>
10 lines
285 B
Python
10 lines
285 B
Python
#!/usr/bin/env python3
|
|
|
|
from deepspeed.nvme import sweep_main, generate_main, parse_sweep_arguments
|
|
|
|
if __name__ == '__main__':
|
|
args = parse_sweep_arguments()
|
|
print(f"Running DeepNVMe performance tuning on {args.nvme_dir}")
|
|
sweep_main(args)
|
|
generate_main(args.log_dir)
|