#!/cs/student/projects3/csml/2024/pchaiyap/environment/bin/python

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)
