fix(cpu_accelerator): 🐛 Convert LOCAL_SIZE to integer (#3971)

Signed-off-by: Javier Salmeron Garcia <[email protected]> Co-authored-by: Jeff Rasley <[email protected]>
deepspeedai · Jul 17, 2023 · f5c834a · f5c834a
1 parent 31ac29d
commit f5c834a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/accelerator/cpu_accelerator.py b/accelerator/cpu_accelerator.py
@@ -40,7 +40,7 @@ def current_device_name(self):
     def device_count(self):
         device_count = int(os.environ.get('LOCAL_SIZE', 0))
         if device_count > 0:
-            return os.environ.get('LOCAL_SIZE')
+            return device_count
         else:
             from deepspeed.utils.numa import get_numa_cores
             # Count NUMA node for number of cpu accelerators. On machine with HBM