@mytec: refactor to ray ready for testing
This commit is contained in:
@@ -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,
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user