Update utils/hardware/device_manager.py

utils/hardware/device_manager.py

@@ -182,6 +182,11 @@ def setup_optimal_threading(self):
                 logger.info(f"Threading already configured: OMP_NUM_THREADS={current_omp}")
                 # Just ensure PyTorch uses the same settings
                 torch.set_num_threads(int(current_omp))
+                
+                # Ensure MKL matches OMP if it's not set
+                if 'MKL_NUM_THREADS' not in os.environ:
+                    os.environ['MKL_NUM_THREADS'] = current_omp
+                
                 return
             
             # Get physical CPU count
@@ -355,4 +360,4 @@ def get_system_diagnostics() -> Dict[str, Any]:
         # Only run setup_optimal_threading if needed (it will check internally)
         manager.setup_optimal_threading()
     except Exception as e:
-        logger.warning(f"Error during device manager initialization: {e}") 
+        logger.warning(f"Error during device manager initialization: {e}")