Add support for rockchip GPU stats (#17812)

* Set rockchip GPU name

* Add support for calculating rockchip GPU load

* Format consistently

* Fix access error
This commit is contained in:
Nicolas Mowen
2025-04-19 16:34:05 -05:00
committed by GitHub
parent 80a13e43e9
commit 38ffe21b69
4 changed files with 35 additions and 2 deletions
+4 -1
View File
@@ -175,6 +175,7 @@ class HailoAsyncInference:
return self.hef.get_input_vstream_infos()[0].shape
def run(self) -> None:
job = None
with self.infer_model.configure() as configured_infer_model:
while True:
batch_data = self.input_store.get()
@@ -202,7 +203,9 @@ class HailoAsyncInference:
bindings_list=bindings_list,
),
)
job.wait(100)
if job is not None:
job.wait(100)
# ----------------- HailoDetector Class ----------------- #