Update utils/hardware/device_manager.py

utils/hardware/device_manager.py

@@ -7,9 +7,16 @@
 import logging
 import platform
 import subprocess
+import os
 from typing import Optional, Dict, Any, List
 from core.exceptions import DeviceError  # Updated import path
 
+# Fix OpenMP threads early with validation
+if 'OMP_NUM_THREADS' not in os.environ:
+    os.environ['OMP_NUM_THREADS'] = '4'
+if 'MKL_NUM_THREADS' not in os.environ:
+    os.environ['MKL_NUM_THREADS'] = '4'
+
 logger = logging.getLogger(__name__)
 
 class DeviceManager:
@@ -412,4 +419,14 @@ def cleanup_device_memory(self):
                 gc.collect()
                 logger.debug("CPU memory cleanup completed")
             except Exception as e:
-                logger.warning(f"CPU memory cleanup failed: {e}") 
+                logger.warning(f"CPU memory cleanup failed: {e}")
+
+# Global instance for singleton pattern
+_device_manager_instance = None
+
+def get_device_manager() -> DeviceManager:
+    """Get or create a singleton DeviceManager instance"""
+    global _device_manager_instance
+    if _device_manager_instance is None:
+        _device_manager_instance = DeviceManager()
+    return _device_manager_instance