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