OpenMP中,任务调度主要用于并行的for循环中,当循环中每次迭代的计算量不相等时,如果简单地给各个线程分配相同次数的迭代的话,会造成各个线程计算负载不均衡,这会使得有些线程先执行完,有些后执行完,造成某些CPU核空闲,影响程序性能。例如以下代码。
void test1()
{
int i, j;
int a[100][100] = {0};
for ( i =0; i < 100; i++)
{
for( j = i; j < 100; j++ )
{
a[i][j] = i*j;
}
}
}
在OpenMP中,对for循环并行化的任务调度使用schedule子句来实现,下面介绍schedule的用法。
schedule的使用格式为:
schedule(type[,size])
schedule有两个参数:type和size,size参数是可选的。
1. type参数
表示调度类型,有四种调度类型如下:
dynamic
guided
runtime
static
这四种调度类型实际上只有static、dynamic、guided三种调度方式,runtime实际上是根据环境变量来选择前三种中的某中类型。
run-sched-var
2. size参数 (可选)
size参数表示循环迭代次数,size参数必须是整数。static、dynamic、guided三种调度方式都可以使用size参数,也可以不使用size参数。当type参数类型为runtime时,size参数是非法的(不需要使用,如果使用的话编译器会报错)。
静态调度(static)
当parallel for编译指导语句没有带schedule子句时,大部分系统中默认采用static调度方式,这种调度方式非常简单。假设有n次循环迭代,t个线程,那么给每个线程静态分配大约n/t次迭代计算。
这里为什么说大约分配n/t次呢?因为n/t不一定是整数,因此实际分配的迭代次数可能存在差1的情况,如果指定了size参数的话,那么可能相差一个size。
静态调度时可以不使用size参数,也可以使用size参数。
不使用size参数时,分配给每个线程的是n/t次连续的迭代,不使用size参数的用法如下:
#pragma omp parallel for schedule(static)
例如以下代码:
void parallel_for_schedule_static1()
{
#pragma omp parallel for schedule(static)
for(int i = 0; i < 10; i++ )
{
printf("i=%d, thread NO=%d\n", i, omp_get_thread_num());
}
}
四核,每个运行2-3次。
使用size参数时,分配给每个线程的size次连续的迭代计算,用法如下:
#pragma omp parallel for schedule(static, N)
例如以下代码:
void parallel_for_schedule_static2()
{
#pragma omp parallel for schedule(static, 2)
for(int i = 0; i < 10; i++ )
{
printf("i=%d, thread NO=%d\n", i, omp_get_thread_num());
}
}
每次指定运行2次。所以线程0运行了4次,其他都是2次。
动态调度(dynamic)
动态调度是动态地将迭代分配到各个线程,动态调度可以使用size参数也可以不使用size参数,不使用size参数时是将迭代逐个地分配到各个线程,使用size参数时,每次分配给线程的迭代次数为指定的size次。
下面为使用动态调度不带size参数的例子:
#pragma omp parallel for schedule(dynamic)
void parallel_for_schedule_dynamic1()
{
#pragma omp parallel for schedule(dynamic)
for(int i = 0; i < 10; i++ )
{
printf("i=%d, thread NO=%d\n", i, omp_get_thread_num());
}
}
任务0-3分配给了线程1-4,之后任务分配给了线程0、2
下面为使用动态调度带size参数的例子:
#pragma omp parallel for schedule(dynamic, N)
void parallel_for_schedule_dynamic2()
{
#pragma omp parallel for schedule(dynamic, 3)
for(int i = 0; i < 10; i++ )
{
printf("i=%d, thread NO=%d\n", i, omp_get_thread_num());
}
}
每3个作一次动态分配
guided调度(guided)
guided调度是一种采用指导性的启发式自调度方法。开始时每个线程会分配到较大的迭代块,之后分配到的迭代块会逐渐递减。迭代块的大小会按指数级下降到指定的size大小,如果没有指定size参数,那么迭代块大小最小会降到1。
下面为使用引导调度不带size参数的例子:
#pragma omp parallel for schedule(guided)
void parallel_for_schedule_guided1()
{
#pragma omp parallel for schedule(guided)
for(int i = 0; i < 20; i++ )
{
printf("i=%2d, thread NO=%d\n", i, omp_get_thread_num());
}
}
分配数量逐渐从5,4,3,2减小。
下面为使用引导调度不带size参数的例子:
#pragma omp parallel for schedule(guided,N)
void parallel_for_schedule_guided2()
{
#pragma omp parallel for schedule(guided, 4)
for(int i = 0; i < 20; i++ )
{
printf("i=%2d, thread NO=%d\n", i, omp_get_thread_num());
}
}
分配数量逐渐从5减小到4。
runtime调度(rumtime)
runtime调度并不是和前面三种调度方式似的真实调度方式,它是在运行时根据环境变量OMP_SCHEDULE来确定调度类型,最终使用的调度类型仍然是上述三种调度方式中的某种。
例如在unix系统中,可以使用setenv命令来设置OMP_SCHEDULE环境变量:
setenv OMP_SCHEDULE “dynamic, 2”
上述命令设置调度类型为动态调度,动态调度的迭代次数为2。
在windows环境中,可以在”系统属性|高级|环境变量”对话框中进行设置环境变量。
#pragma omp parallel for schedule(runtime)
void parallel_for_schedule_runtime()
{
#pragma omp parallel for schedule(runtime)
for(int i = 0; i < 10; i++ )
{
printf("i=%2d, thread NO=%d\n", i, omp_get_thread_num());
}
}