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

Signed-off-by: Javier Salmeron Garcia <[email protected]>
Co-authored-by: Jeff Rasley <[email protected]>

Author: javsalgar

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