def setUp(self):
self.config = HardwareConfig(cpu_ids=[0, 1, 2, 3])
self._default_options: cg.GenerateOptions = cg.GenerateOptions(
generate_exhaustive_options=True, max_sims_per_gpu=4)
def testMinimalSubset(self):
self.maxDiff = None
self.config.gpu_ids = [0]
options = cg.GenerateOptions(max_sims_per_gpu=2,
generate_exhaustive_options=False)
result = cg.createRunOptionsForSingleConfig(self.config, self.version,
options)
expected = [
# PME = gpu cases need npme=1
{
**self.expected_base, "ntmpi": 4,
"ntomp": 1,
"nb": "gpu",
"bonded": "gpu",
"pme": "gpu",
"npme": 1,
"update": "gpu",
"gputasks": "0000"
},
{
**self.expected_base, "ntmpi": 2,
"ntomp": 2,
"nb": "gpu",
"bonded": "gpu",
"pme": "gpu",
"npme": 1,
"update": "gpu",
"gputasks": "00"
},
# Note the double gputask even though there's only one rank
{
**self.expected_base, "ntmpi": 1,
"ntomp": 4,
"nb": "gpu",
"bonded": "gpu",
"pme": "gpu",
"update": "gpu",
"gputasks": "00"
},
# update = cpu cases
{
**self.expected_base, "ntmpi": 4,
"ntomp": 1,
"nb": "gpu",
"pme": "cpu",
"bonded": "cpu",
"update": "cpu",
"gputasks": "0000"
},
{
**self.expected_base, "ntmpi": 2,
"ntomp": 2,
"nb": "gpu",
"bonded": "cpu",
"pme": "cpu",
"update": "cpu",
"gputasks": "00"
},
{
**self.expected_base, "ntmpi": 1,
"ntomp": 4,
"nb": "gpu",
"bonded": "cpu",
"pme": "cpu",
"update": "cpu",
"gputasks": "0"
},
]
self.assertCountEqual(result, expected)
def setUp(self) -> None:
self._default_options: cg.GenerateOptions = cg.GenerateOptions(
generate_exhaustive_options=True, max_sims_per_gpu=4)