(self, num_gpu=0, num_cpu=0, mem=0)
| 31 | utils.print_fn(' Node[%d] has %d gpus, %d cpus, %d G memory' % (id, num_gpu, num_cpu, mem)) |
| 32 | |
| 33 | def init_node(self, num_gpu=0, num_cpu=0, mem=0): |
| 34 | if num_gpu != 0: |
| 35 | self.num_gpu = num_gpu |
| 36 | self.free_gpus = num_gpu |
| 37 | if num_cpu != 0: |
| 38 | self.num_cpu = num_cpu |
| 39 | self.free_cpus = num_cpu |
| 40 | if mem != 0: |
| 41 | self.mem = mem |
| 42 | self.free_mem = mem |
| 43 | self.gpu_job_list = [{0:[], 1:[]} for i in range(self.num_gpu)] |
| 44 | self.gpu_util_list = [0.0 for i in range(self.num_gpu)] |
| 45 | |
| 46 | self.add_gpus(self.num_gpu) |
| 47 | self.add_cpus(self.num_gpu) |
| 48 | |
| 49 | |
| 50 | ''' GPU ''' |
no test coverage detected