frozenleaves/DeepSpeed
1
0
Fork 0
mirror of https://github.com/deepspeedai/DeepSpeed.git synced 2025-10-20 15:33:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

Use accelerator pin memory

Browse Source 
Signed-off-by: Olatunji Ruwase <tjruwase@gmail.com>
This commit is contained in:
Olatunji Ruwase
2025-07-07 15:07:59 +00:00
parent 15f054d97f
commit 970955dcac
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
deepspeed/nvme/perf_run_sweep.py
View File

@ -24,7 +24,7 @@ DEFAULT_SWEEP_CONFIG = {
"queue_depth": [32, 128],
"sequential_requests": [True, False],
"single_submit": [False, True],
"io_parallel": [1, 2, 4, 8],
"io_parallel": [1, 8],
}
@ -41,7 +41,7 @@ class SweepConfig(object):
self.verbose = args.verbose
self.other_options = f'{OTHER_OPTIONS} --loops {args.loops} --io_size {args.io_size}'
if args.gpu:
self.other_options += ' --gpu'
self.other_options += ' --gpu --use_accelerator_pin_memory'
if args.gds:
self.other_options += ' --use_gds'