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 _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)
