(self, id, ip)
| 30 | |
| 31 | class Node: |
| 32 | def __init__(self, id, ip): |
| 33 | self.id = id |
| 34 | self.ip = ip |
| 35 | self.job_run_list = [] |
| 36 | self.job_run_list_lock = threading.Lock() |
| 37 | self.gpu_list = [] |
| 38 | |
| 39 | r = redis_connection() |
| 40 | config = get_config_file() |
| 41 | self.gpu_num = config['ServerConfig']['serverGPU_'] |
| 42 | for i in range(self.gpu_num): |
| 43 | redis_key = f"{self.ip}:{i}" |
| 44 | gpu_info_pro = r.hgetall(redis_key) |
| 45 | gpu = GPU_info(gpu_info_pro) |
| 46 | self.gpu_list.append(gpu) |
| 47 | print(gpu) |
| 48 | |
| 49 | |
| 50 | def update_gpu_info(self, gpu_id, job, type):#type=0:分配内存，type=1:释放内存 |
nothing calls this directly
no test coverage detected