Skip to content
Snippets Groups Projects

Resolve "Flavor for compute backend config for local runs"

Merged Omar Ahmed requested to merge 203-flavor-for-compute-backend-config-for-local-runs into main
Files
2
+ 7
9
@@ -18,7 +18,7 @@ import mantik.utils.env_vars as env_vars
logger = logging.getLogger(__name__)
def get_gpu_info() -> dict:
def get_gpu_info() -> list:
import GPUtil
gpus = GPUtil.getGPUs()
@@ -27,10 +27,10 @@ def get_gpu_info() -> dict:
for gpu in gpus:
gpu_info = {
"name": gpu.name,
"id": str(gpu.id),
"driver": gpu.driver,
"memory_total": str(gpu.memoryTotal),
"Name": gpu.name,
"ID": str(gpu.id),
"Driver": gpu.driver,
"TotalMemory": str(gpu.memoryTotal),
}
gpu_info_list.append(gpu_info)
@@ -44,7 +44,7 @@ def get_system_details() -> dict:
system_details = {
"Platform": platform.platform(),
"CpuModel": cpuinfo.get_cpu_info()["brand_raw"], # FIX
"CpuModel": cpuinfo.get_cpu_info().get("brand_raw"),
"CpuCoreCount": psutil.cpu_count(),
"TotalMemoryGB": psutil.virtual_memory().total / (1024**3),
"GPUInfo": get_gpu_info(),
@@ -184,9 +184,7 @@ def run_project(
mantik_token = mantik.authentication.auth.get_valid_access_token()
system_details = {}
if backend_config_system_info:
system_details = get_system_details()
system_details = get_system_details() if backend_config_system_info else {}
local_runs.run(
data=schemas.RunConfiguration(
Loading