From 7764a6a7f528b88d9a547378e1ca62701672590c Mon Sep 17 00:00:00 2001 From: yuruofeifei Date: Sun, 19 Apr 2026 01:40:28 -0700 Subject: [PATCH] =?UTF-8?q?Fix=20typo:=20pin=5Femory=20=E2=86=92=20pin=5Fm?= =?UTF-8?q?emory=20in=20PrefetchMuxInputMultiProcessQueue?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-Authored-By: Claude Opus 4.6 (1M context) --- steptronoss/data/nextable/async_accelerate.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/steptronoss/data/nextable/async_accelerate.py b/steptronoss/data/nextable/async_accelerate.py index 842afa68..8e4f141b 100644 --- a/steptronoss/data/nextable/async_accelerate.py +++ b/steptronoss/data/nextable/async_accelerate.py @@ -41,7 +41,7 @@ def __init__( self.input_queues = [mp.Queue(single_queue_size) for _ in range(mux_size)] self.mux_size = len(self.input_queues) self.prefetch_output_queue: queue.Queue = queue.Queue(PREFETCH_QUEUE_SIZE) - self.pin_emory = pin_memory + self.pin_memory = pin_memory self.idx = -1 self.thread_stop_signal = False self._last_prefetch_empty_warn = 0.0 @@ -85,7 +85,7 @@ def _prefetch_loop(self) -> None: try: while not self.thread_stop_signal: data = self._get_one_data() - if self.pin_emory and not isinstance(data, Exception): + if self.pin_memory and not isinstance(data, Exception): # Pin tensors to speed up host->device transfers. data = pin_memory(data) while not self.thread_stop_signal: