Added deviceID to printed name for GPUs

这个提交包含在:
craig-warren
2023-12-09 23:56:34 +00:00
父节点 591a6e63e3
当前提交 0a37f23ef5
共有 2 个文件被更改,包括 4 次插入2 次删除

查看文件

@@ -91,6 +91,7 @@ class ModelConfig:
deviceID = 0
self.device = {"dev": sim_config.set_model_device(deviceID), "snapsgpu2cpu": False}
self.device["deviceID"] = deviceID
# Total memory usage for all grids in the model. Starts with 50MB overhead.
self.mem_overhead = 65e6

查看文件

@@ -309,11 +309,12 @@ class ModelBuildRun:
if config.sim_config.general["solver"] == "opencl":
solvername = "OpenCL"
platformname = " on " + " ".join(config.get_model_config().device["dev"].platform.name.split())
devicename = " ".join(config.get_model_config().device["dev"].name.split())
else:
solvername = "CUDA"
platformname = ""
devicename = " ".join(config.get_model_config().device["dev"].name().split())
devicename = (f'[{config.get_model_config().device["deviceID"]}]'
f'{" ".join(config.get_model_config().device["dev"].name().split())}')
logger.basic(
f"\nModel {config.model_num + 1}/{config.sim_config.model_end} "
f"on {config.sim_config.hostinfo['hostname']} "