Update BUFSIZE to come from autotuner's constants.py, not numpy (#5686)

microsoft · Jun 19, 2024 · 0c979d6 · 0c979d6
1 parent 4000cee
commit 0c979d6
Showing 1 changed file with 1 addition and 6 deletions.
diff --git a/deepspeed/autotuning/scheduler.py b/deepspeed/autotuning/scheduler.py
@@ -5,11 +5,6 @@
 
 import copy
 
-import numpy
-if numpy.__version__ < '2.0.0':
-    from numpy import BUFSIZE
-else:
-    from numpy._core.umath import BUFSIZE
 import json
 import subprocess
 import sys
@@ -22,7 +17,7 @@
 from tqdm import tqdm
 
 from ..utils import logger
-from .constants import AUTOTUNING, AUTOTUNING_METRIC_PATH
+from .constants import AUTOTUNING, AUTOTUNING_METRIC_PATH, BUFSIZE
 from .utils import get_val_by_key, search_error, was_interruptted
 """
 thread-0: loop over experiment queue dispatching experiments if they become available