def _prepare_pbs_handlers(self, id_string): cluster = util.gpufile_to_clusterspec(os.environ['PBS_GPUFILE'], self.config.cluster_interface) def _start_socketserver(addr, port): return subprocess.Popen(['pbsdsh', '-h', addr, 'sh', '-c', ". %s ; python %s/socketserver.py :%s %s" % (self.config.cluster_pbs_initscript, os.path.realpath(os.path.dirname(util.__file__)), port, id_string)]) return cluster, _start_socketserver
def test_gpufile_processing(self): fd, path = tempfile.mkstemp() f = os.fdopen(fd, 'w') f.write("node1-gpu2.domain-gpu0\n" "node1-gpu2.domain-gpu2\n" "node2-gpu4.domain-gpu0") f.close() random.seed(1234) cluster = util.gpufile_to_clusterspec(path) os.unlink(path) self.assertEqual( [config.MachineSpec('socket=node1-gpu2.domain:%d' % PORT, 'node1-gpu2.domain', [0, 2], None), config.MachineSpec('socket=node2-gpu4.domain:%d' % PORT, 'node2-gpu4.domain', [0], None)], cluster.nodes)
def test_gpufile_processing(self): fd, path = tempfile.mkstemp() f = os.fdopen(fd, 'w') f.write("node1-gpu2.domain-gpu0\n" "node1-gpu2.domain-gpu2\n" "node2-gpu4.domain-gpu0") f.close() random.seed(1234) cluster = util.gpufile_to_clusterspec(path) os.unlink(path) self.assertEqual([ config.MachineSpec('socket=node1-gpu2.domain:%d' % PORT, 'node1-gpu2.domain', [0, 2], None), config.MachineSpec('socket=node2-gpu4.domain:%d' % PORT, 'node2-gpu4.domain', [0], None) ], cluster.nodes)