administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-04-03 22:04:53 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Simplify handling

Browse Source
This commit is contained in:
Nicolas Mowen 2026-03-28 10:46:52 -06:00
parent 31bd461ece
commit d03a24b8fe
1 changed files with 28 additions and 32 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

58
frigate/util/services.py
View File

@ -347,8 +347,10 @@ def get_intel_gpu_stats(intel_gpu_device: Optional[str]) -> Optional[dict[str, s
results: dict[str, str] = {} results: dict[str, str] = {}
rc6_values = [] rc6_values = []
render = {"global": []} render_global = []
video = {"global": []} video_global = []
# per-client: {pid: [total_busy_per_sample, ...]}
client_usages: dict[str, list[float]] = {}
for block in data: for block in data:
# rc6 residency: percentage of time GPU is idle # rc6 residency: percentage of time GPU is idle
@ -363,31 +365,28 @@ def get_intel_gpu_stats(intel_gpu_device: Optional[str]) -> Optional[dict[str, s
video_frame = global_engine.get("Video/0", {}).get("busy") video_frame = global_engine.get("Video/0", {}).get("busy")
if render_frame is not None: if render_frame is not None:
render["global"].append(float(render_frame)) render_global.append(float(render_frame))
if video_frame is not None: if video_frame is not None:
video["global"].append(float(video_frame)) video_global.append(float(video_frame))
clients = block.get("clients", {}) clients = block.get("clients", {})
if clients and len(clients): if clients:
for client_block in clients.values(): for client_block in clients.values():
key = client_block["pid"] pid = client_block["pid"]
if render.get(key) is None: if pid not in client_usages:
render[key] = [] client_usages[pid] = []
video[key] = []
client_engine = client_block.get("engine-classes", {}) # Sum all engine-class busy values for this client
total_busy = 0.0
for engine in client_block.get("engine-classes", {}).values():
busy = engine.get("busy")
if busy is not None:
total_busy += float(busy)
render_frame = client_engine.get("Render/3D", {}).get("busy") client_usages[pid].append(total_busy)
video_frame = client_engine.get("Video", {}).get("busy")
if render_frame is not None:
render[key].append(float(render_frame))
if video_frame is not None:
video[key].append(float(video_frame))
# Overall GPU usage from rc6 (idle) residency # Overall GPU usage from rc6 (idle) residency
if rc6_values: if rc6_values:
@ -397,24 +396,21 @@ def get_intel_gpu_stats(intel_gpu_device: Optional[str]) -> Optional[dict[str, s
results["mem"] = "-%" results["mem"] = "-%"
# Encoder: Render/3D engine (compute/shader encode) # Encoder: Render/3D engine (compute/shader encode)
if render["global"]: if render_global:
results["enc"] = ( results["enc"] = f"{round(sum(render_global) / len(render_global), 2)}%"
f"{round(sum(render['global']) / len(render['global']), 2)}%"
)
# Decoder: Video engine (fixed-function codec) # Decoder: Video engine (fixed-function codec)
if video["global"]: if video_global:
results["dec"] = f"{round(sum(video['global']) / len(video['global']), 2)}%" results["dec"] = f"{round(sum(video_global) / len(video_global), 2)}%"
if len(render.keys()) > 1: # Per-client GPU usage (sum of all engines per process)
if client_usages:
results["clients"] = {} results["clients"] = {}
for key in render.keys(): for pid, samples in client_usages.items():
if key == "global" or not render[key] or not video[key]: if samples:
continue results["clients"][pid] = (
f"{round(sum(samples) / len(samples), 2)}%"
results["clients"][key] = (
f"{round(((sum(render[key]) / len(render[key])) + (sum(video[key]) / len(video[key]))) / 2, 2)}%"
) )
return results return results