def test_more_nodes_than_subdomains(self):
nodes = [
MachineSpec('a', 'a', gpus=[0]),
MachineSpec('b', 'b', gpus=[1])
]
subds = [
SubdomainSpec2D((0,0), (10, 10), id_=0),
]
assignments = controller.split_subdomains_between_nodes(nodes, subds)
self.assertEqual(assignments, [[subds[0]]])
def test_more_nodes_than_subdomains(self):
nodes = [
MachineSpec('a', 'a', gpus=[0]),
MachineSpec('b', 'b', gpus=[1])
]
subds = [
SubdomainSpec2D((0, 0), (10, 10), id_=0),
]
assignments = controller.split_subdomains_between_nodes(nodes, subds)
self.assertEqual(assignments, [[subds[0]]])
def test_1_1_mapping_multigpi(self):
nodes = [
MachineSpec('a', 'a', gpus=[0, 1]),
MachineSpec('b', 'b', gpus=[1])
]
subds = [
SubdomainSpec2D((0,0), (10, 10), id_=0),
SubdomainSpec2D((0,10), (10, 10), id_=1),
SubdomainSpec2D((0,20), (10, 10), id_=2)
]
assignments = controller.split_subdomains_between_nodes(nodes, subds)
self.assertEqual(assignments, [[subds[0], subds[1]], [subds[2]]])
def test_1_1_mapping_multigpi(self):
nodes = [
MachineSpec('a', 'a', gpus=[0, 1]),
MachineSpec('b', 'b', gpus=[1])
]
subds = [
SubdomainSpec2D((0, 0), (10, 10), id_=0),
SubdomainSpec2D((0, 10), (10, 10), id_=1),
SubdomainSpec2D((0, 20), (10, 10), id_=2)
]
assignments = controller.split_subdomains_between_nodes(nodes, subds)
self.assertEqual(assignments, [[subds[0], subds[1]], [subds[2]]])
