OpenMP编程的任务调度控制

在OpenMP的for任务分担中,各个线程的任务划分是可以由程序员控制调整的。考虑这样一种情况,当在一个循环中每次迭代的计算量不相等时,如果根据系统默认简单的给每个线程分配相同次数的迭代量的话,会导致有些线程先执行玩,有些线程后执行完,造成CPU核的空闲,降低程序的运行效率。这种情况下就有必要人为的对各个线程的任务划分进行分配。


例如对如下的循环:

#pragma omp parallel for 
	for (int i = 0; i < 100; i++)
	{
		cout << i*i << endl;
	}

显然最后几次的计算量是远远大于前几次的计算量的,解决这种由于各个线程间均分迭代次数造成的负载不平衡的问题,可以通过schedule子句的使用来避免。


schedule子句的用法


使用格式:

schedule(type,size)

type表示调度类型,共有4种类型可选:

  • static
  • dynamic
  • guided
  • runtime

size参数定义了迭代次数最小的划分单位,每个线程依次分配size个迭代次数。


static静态调度


#include<iostream>
#include"omp.h"

using namespace std;

void main()
{
#pragma omp parallel for schedule(static,3)
	for (int i = 0; i < 9; i++)
	{
		printf("i=%d, thread_id=%d\n", i, omp_get_thread_num());
	}
	system("pause");
}

输出:



从打印结果可以看出每个线程依次分配到3个连续的迭代计算。


dynamic动态调度


动态调度是动态的将迭代分配到各个线程上,可以使用size参数也可以不使用size参数,不适用size参数情况下size值默认为1,根据每个线程的完成情况每次再分配2个迭代给已完成迭代任务的线程,使用size参数则每次分配size个迭代次数。较快迭代完成的线程可能执行更多次迭代,较慢的线程执行较少的迭代,以此来解决各线程间负载分配不均衡的问题。


示例:

#include<iostream>
#include"omp.h"

using namespace std;

void main()
{
#pragma omp parallel for schedule(dynamic,2)
	for (int i = 0; i < 12; i++)
	{
		printf("i=%d, thread_id=%d\n", i*i*i*i, omp_get_thread_num());
	}
	system("pause");
}

输出:



执行乘法运算,i比较小的时候计算量小,所以可以看到编号为0的线程执行了6次,编号为1、2和3的线程分别执行了2次。


guided调度


guided调度跟dynamic动态调度类似,是一种采用指导性的启发式自调度方法。刚开始时候每个线程会分配到较大的迭代块,之后每次分配到的迭代块会以指数级下降,直到下降到自定义的size大小,如果没有定义size,则默认为1。


runtime调度


runtime调度根据运行时的环境变量OMP_SCHEDULE来确定调度类型,最后还是会落实到schedule、dynamic和guided中的一种上来。

例如在unix系统中,可以使用setenv命令来设置OMP_SCHEDULE环境变量:

  setenv OMP_SCHEDULE “dynamic, 2”  

 如果程序中选择runtime调度,那么上述命令设置调度类型为动态调度,动态调度的size为2。在windows环境中,可以在“系统属性|高级|环境变量”对话框中进行设置环境变量。


sections制导指令


for制导指令用于迭代计算的任务分配,sections制导指令用于非迭代计算的任务分配,sections的语法格式如下:

#pragma omp parallel sections[子句]


#include<iostream>
#include"omp.h"

using namespace std;

void main()
{
#pragma omp parallel sections
	{
#pragma omp section
		{
			for (int i = 0; i < 4; i++)
			{
				printf("i=%d, thread_id=%d\n", i, omp_get_thread_num());
			}
		}
#pragma omp section
		{

			for (int i = 0; i < 4; i++)
			{
				printf("j=%d, thread_id=%d\n", i, omp_get_thread_num());
			}
		}
	}
	system("pause");
}

输出:



sections语句块中一共有2个section段落,所以生成了2个线程并行执行。


single制导指令


在parallel制导指令产生的并行域中的语句,一般会默认生成4个不同的线程对相同的语句分别执行一次,如果想要其中某些操作只执行一次,可以使用single制导指令,其后的语句块只会被单个线程执行。

#include<iostream>
#include"omp.h"

using namespace std;

void main()
{
#pragma omp parallel
	{
#pragma omp single
		{
			printf("single thread_id=%d\n", omp_get_thread_num());
		}

		printf("多线程执行,thread_id=%d\n", omp_get_thread_num());
	}
	system("pause");
}

输出:



由single引导的语句块只执行了一次,第二个printf语句没有single指令,所以被4个不同的线程分别执行了一次。


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值