@mytec: refactor to ray ready for testing

This commit is contained in:
2026-01-31 21:09:10 +02:00
parent 3b010fed83
commit 221000d5b3
6 changed files with 278 additions and 154 deletions

View File

@@ -6,9 +6,22 @@ router = APIRouter()
@router.get("/info")
async def get_system_info():
"""Return system info: CPU cores, GPU availability, parallel support."""
"""Return system info: CPU cores, GPU availability, parallel backend."""
cpu_cores = mp.cpu_count() or 1
# Check Ray
ray_available = False
ray_initialized = False
try:
from app.services.parallel_coverage_service import RAY_AVAILABLE
ray_available = RAY_AVAILABLE
if ray_available:
import ray
ray_initialized = ray.is_initialized()
except Exception:
pass
# Check GPU
gpu_info = None
try:
import cupy as cp
@@ -24,7 +37,9 @@ async def get_system_info():
return {
"cpu_cores": cpu_cores,
"parallel_workers": min(cpu_cores, 14),
"parallel_enabled": True,
"parallel_backend": "ray" if ray_available else "sequential",
"ray_available": ray_available,
"ray_initialized": ray_initialized,
"gpu": gpu_info,
"gpu_enabled": gpu_info is not None,
}