From f5c834a6e0eedd440af85b0f577ca9c8f5624630 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Javier=20J=2E=20Salmer=C3=B3n-Garc=C3=ADa?=
 <jsalmeron@vmware.com>
Date: Mon, 17 Jul 2023 19:44:40 +0200
Subject: [PATCH] fix(cpu_accelerator): :bug: Convert LOCAL_SIZE to integer
 (#3971)

Signed-off-by: Javier Salmeron Garcia <jsalmeron@vmware.com>
Co-authored-by: Jeff Rasley <jerasley@microsoft.com>
---
 accelerator/cpu_accelerator.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/accelerator/cpu_accelerator.py b/accelerator/cpu_accelerator.py
index b1fb036f..765f4d15 100644
--- a/accelerator/cpu_accelerator.py
+++ b/accelerator/cpu_accelerator.py
@@ -40,7 +40,7 @@ class CPU_Accelerator(DeepSpeedAccelerator):
     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
-- 
GitLab