本文整理汇总了Python中job.Job.setTasksPerNode方法的典型用法代码示例。如果您正苦于以下问题：Python Job.setTasksPerNode方法的具体用法？Python Job.setTasksPerNode怎么用？Python Job.setTasksPerNode使用的例子？那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类job.Job的用法示例。
在下文中一共展示了Job.setTasksPerNode方法的1个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: DistributionTestCase
# 需要导入模块: from job import Job [as 别名]
# 或者: from job.Job import setTasksPerNode [as 别名]
class DistributionTestCase(unittest.TestCase):
def setUp(self):
self.job = Job()
self.batch = Batch()
self.resource = Resource()
rootDir = os.environ['BOLT_DIR']
self.batch.readConfig(rootDir + configDir + "/" + batchConfig)
self.resource.readConfig(rootDir + configDir + "/" + resourceConfig)
def testParallelTaskDitributionPureMPI(self):
"""Pure MPI task distribution (fully populated)."""
# Set the parallel distribution
self.job.setTasks(1024)
self.job.setTasksPerNode(self.resource.numCoresPerNode())
self.job.setThreads(1)
self.job.setParallelDistribution(self.resource, self.batch)
correct = "aprun -n 1024 -N 32 -S 8 -d 1"
self.assertEqual(self.job.runLine, correct, "Value= '{0}', Expected= '{1}'".format(self.job.runLine, correct))
def testParallelTaskDitributionHalfPopulate(self):
"""Pure MPI task distribution (half populated)."""
# Set the parallel distribution
self.job.setTasks(1024)
self.job.setTasksPerNode(16)
self.job.setThreads(1)
self.job.setParallelDistribution(self.resource, self.batch)
correct = "aprun -n 1024 -N 16 -S 4 -d 2"
self.assertEqual(self.job.runLine, correct, "Value= '{0}', Expected= '{1}'".format(self.job.runLine, correct))
def testParallelTaskDitributionTwoThreads(self):
"""Hybrid MPI/OpenMP task distribution (2 OpenMP threads)."""
# Set the parallel distribution
self.job.setTasks(1024)
self.job.setTasksPerNode(16)
self.job.setThreads(2)
self.job.setParallelDistribution(self.resource, self.batch)
correct = "export OMP_NUM_THREADS=2\naprun -n 1024 -N 16 -S 4 -d 2"
self.assertEqual(self.job.runLine, correct, "Value= '{0}', Expected= '{1}'".format(self.job.runLine, correct))
def testParallelTaskDitributionThreeThreads(self):
"""Hybrid MPI/OpenMP task distribution (3 OpenMP threads)."""
# Set the parallel distribution
self.job.setTasks(1024)
self.job.setTasksPerNode(10)
self.job.setThreads(3)
self.job.setParallelDistribution(self.resource, self.batch)
correct = "export OMP_NUM_THREADS=3\naprun -n 1024 -N 10 -d 3"
self.assertEqual(self.job.runLine, correct, "Value= '{0}', Expected= '{1}'".format(self.job.runLine, correct))