From 68e87a5beeb02a2ec179a20ccd0cbc746da353ca Mon Sep 17 00:00:00 2001 From: Alexander Thiess Date: Wed, 10 Sep 2025 05:10:51 +0200 Subject: [PATCH] gpu info definition --- src/services/__init__.py | 3 ++- src/services/gpu.py | 8 ++++++++ 2 files changed, 10 insertions(+), 1 deletion(-) create mode 100644 src/services/gpu.py diff --git a/src/services/__init__.py b/src/services/__init__.py index 591317a..bf1ac69 100644 --- a/src/services/__init__.py +++ b/src/services/__init__.py @@ -1,3 +1,4 @@ from .model_manager import ModelManager +from gpu import get_gpu_info -__all__ = ["ModelManager"] \ No newline at end of file +__all__ = ["ModelManager", "get_gpu_info"] diff --git a/src/services/gpu.py b/src/services/gpu.py new file mode 100644 index 0000000..70f6489 --- /dev/null +++ b/src/services/gpu.py @@ -0,0 +1,8 @@ +from typing import Dict + + +def get_gpu_info() -> Dict: + # use the command "amdgpu_top -d -J" + # it will return the gpu information + # this function should just return all information of the one gpu it will find. we just expect it to be one gpu. no handling of multiple ones. + ...