From 902e78c989383dac09d47325f39b5c83d5e7f889 Mon Sep 17 00:00:00 2001 From: digger yu Date: Mon, 1 Sep 2025 09:12:40 +0800 Subject: [PATCH] fix typo s/1014 /1024 (#7528) fix typo s/1014 /1024 s/was_interruptted /was_interrupted detail info modified: deepspeed/autotuning/scheduler.py modified: deepspeed/autotuning/utils.py Co-authored-by: Olatunji Ruwase --- deepspeed/autotuning/scheduler.py | 4 ++-- deepspeed/autotuning/utils.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/deepspeed/autotuning/scheduler.py b/deepspeed/autotuning/scheduler.py index 35f709ead..14e9541d0 100755 --- a/deepspeed/autotuning/scheduler.py +++ b/deepspeed/autotuning/scheduler.py @@ -18,7 +18,7 @@ from tqdm import tqdm from ..utils import logger from .constants import AUTOTUNING, AUTOTUNING_METRIC_PATH, BUFSIZE -from .utils import get_val_by_key, search_error, was_interruptted +from .utils import get_val_by_key, search_error, was_interrupted """ thread-0: loop over experiment queue dispatching experiments if they become available thread-N: start each experiment in its own thread @@ -76,7 +76,7 @@ class ResourceManager: # skip existing experiments (except for the ones that were interrupted) if os.path.exists(result_dir) and os.path.exists(stderr_file): - if not was_interruptted(stderr_file): + if not was_interrupted(stderr_file): err = search_error(stderr_file) exp_id = exp["exp_id"] self.finished_experiments[exp_id] = (exp, err) diff --git a/deepspeed/autotuning/utils.py b/deepspeed/autotuning/utils.py index b85135352..53d5d68ac 100644 --- a/deepspeed/autotuning/utils.py +++ b/deepspeed/autotuning/utils.py @@ -26,7 +26,7 @@ def search_error(filename): return None -def was_interruptted(filename): +def was_interrupted(filename): if not os.path.exists(filename): return "stderr.log does not exist" with open(filename) as f: @@ -422,7 +422,7 @@ def memory_to_string(n, postfix="", units=None, precision=2): elif n // 10**6 > 0: return str(round(n / 1024**2, precision)) + " M" + postfix elif n // 10**3 > 0: - return str(round(n / 1014, precision)) + " K" + postfix + return str(round(n / 1024, precision)) + " K" + postfix else: return str(n) + " " else: