GPU dict update

这个提交包含在:
craig-warren
2023-12-10 00:01:23 +00:00
父节点 0a37f23ef5
当前提交 8235cb37b3

查看文件

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