C++ AMP 介绍(二)

来源:互联网 发布:大连 旅游 知乎 编辑:程序博客网 时间:2024/05/24 02:51

C++ AMP 介绍(二)

最后更新日期:2014-05-02

阅读前提:《C++ AMP介绍(一)》

环境:Windows 8.1 64bit英文版,Visual Studio 2013 Update1英文版,Nvidia QuadroK600 显卡

内容简介

         介绍C++ AMP的 array、array_view、extent类和平铺的知识。

正文

数据的移动

         array和 array_view两个数据容器(模板类)用于把数据从运行时库(CPU)移到加速器(显卡或通用计算卡)上,array类在构造时建立数据的深拷贝,把数据复制到加速器(GPU)上,而array_view类是个包装类,仅仅当核心函数(kernel function)要用到数据时,才把源数据复制到加速器上。

#include <amp.h>using namespace concurrency;//演示array类的使用方式void test_array(){//测试数据std::vector<int> data(5);for (int count = 0; count < 5; count++){data[count] = count;}//构造array实例array<int, 1> a(5, data.begin(), data.end());parallel_for_each(a.extent,[=, &a](index<1> idx) restrict(amp){a[idx] = a[idx] * 10;});//array实例a不需要调用同步方法//但是需要赋值给datadata = a;//输出0,10,20,30,40for (int i = 0; i < 5; i++){std::cout << data[i] << "\n";}}


        array_view同array之间几乎有相同的成员,但是它们底层的行为不一样,所以当你建立两个指向同一个数据源的array_view实例时,实际上它们指向同一个内存地址,数据只有当需要的时候才会被复制到加速器中,所以你得注意数据的同步,array_view类的主要好处是数据仅当要被加速器用到的时候才会被移动。

   共享内存是能被CPU和GPU访问的内存,array类可以控制共享内存的存取方式,但是首先我们需要测试加速器是不是支持共享内存,下面是array使用共享内存的示例代码。

int test_sharedMemory(){// 一台计算机中可能有多块加速器,取默认加速器accelerator acc = accelerator(accelerator::default_accelerator);// 测试默认加速器是否支持共享内存if (!acc.supports_cpu_shared_memory){std::cout << "The default accelerator does not support shared memory" << std::endl;return 1;}// 设置cpu默认存取方式acc.set_default_cpu_access_type(access_type_read_write);//为acc加速器建立accelerator_view(加速器视图)实例//读写方式默认为加速器default_cpu_access_type属性的设定accelerator_view acc_v = acc.default_view;// extent指示array实例建立一个含10个元素的一维数组extent<1> ex(10);// 指定加速器视图,输入数组在CPU上只写array<int, 1> arr_w(ex, acc_v, access_type_write);// 指定加速器视图,输出数组在CPU上只读array<int, 1> arr_r(ex, acc_v, access_type_read);// 指定加速器视图,可以在CPU上读写的数组array<int, 1> arr_rw(ex, acc_v, access_type_read_write);return 0;}


 

index 类

   index类指定元素在array或array_view对象中的位置,下面是index类的使用示例代码

void test_indexClass(){int aCPP[] = { 1, 2, 3,4, 5, 6 };//新建2维(两行三列)array_view包装器array_view<int, 2> a(2, 3, aCPP);index<2> idx(1, 2);//输出6std::cout << a[idx] << "\n";}


 

extent类

         虽然extent类在很多场合下不是必要的,但是微软的部分示例代码使用到了extent class,所以有必要介绍下extent class。

         extentclass用来指定array或array_view各个维度的元素数量,你可以使用extent class建立array或array_view对象,也可以从array或array_view对象中存取extent,下面的例子演示了extent class的使用。

void test_extentClass(){int aCPP[] = { 111, 112, 113, 114,            121, 122, 123, 124,    131, 132, 133, 134,   211, 212, 213, 214,   221, 222, 223, 224,   231, 232, 233, 234 };extent<3> e(2, 3, 4);array_view<int, 3> a(e, aCPP);//断言extent[0],[1],[2]的属性分别为2、3、4assert(2 == a.extent[0]);assert(3 == a.extent[1]);assert(4 == a.extent[2]);}


parallel_for_each函数

         我们在上篇文章中调用过parallel_for_each函数,它有两个入口参数,第一个入口参数为计算域,是个extent或tiled_extent对象,定义了要在加速器上并发运行的线程集合,它会为每个元素生成一根用于计算的线程。第二个参数是lambda表达式,定义了要在每根线程上运行的代码。

 

加速代码:  砖面(Tiles )和边界(Barriers)

   将全体线程划分为若干个具有相等数量矩形(M*N根)线程集合,每个集合称为tile(砖面),多个tile(砖面)组成全体线程,叫做平铺(tiling)。

   若要使用平铺,在parallel_for_each 方法中的计算域上调用 extent::tile 方法,并在 lambda 表达式中使用 tiled_index 对象。

         下面是两张来自微软官网的砖面(tile)的组织图,可以看到如何索引元素。

图中的idx是index类,sample是全局空间(array或array_view对象)

下图中的t_idx是index类,descriptions是全局空间(array或array_view对象)



下面这个来自微软官方的例子,每2*2=4根线程组成一个砖面(tile),计算砖面(tile)中元素的平均值。

void test_tile(){// 测试样本:int sampledata[] = {2, 2, 9, 7, 1, 4,4, 4, 8, 8, 3, 4,1, 5, 1, 2, 5, 2,6, 8, 3, 2, 7, 2 };// The tiles(下面是6个砖面):// 2 2    9 7    1 4// 4 4    8 8    3 4//// 1 5    1 2    5 2// 6 8    3 2    7 2// averagedata用来存放运算结果:int averagedata[] = {0, 0, 0, 0, 0, 0,0, 0, 0, 0, 0, 0,0, 0, 0, 0, 0, 0,0, 0, 0, 0, 0, 0,};//每四个元素(四根线程)组成一个tile(砖面),所以共有六个tile(砖面)array_view<int, 2> sample(4, 6, sampledata);array_view<int, 2> average(4, 6, averagedata);//通过[1]extent.tile代替extent[2]tiled_index代替index,启用平铺模式parallel_for_each(// 把extent切分为以2*2为单位的tile(砖面)sample.extent.tile<2, 2>(),[=](tiled_index<2, 2> idx) restrict(amp){//tile_static关键字的变量范围是整个tile(砖面)//所以每个tile(砖面)(2*2=4根线程)只实例化一个tile_statictile_static int nums[2][2];//tile(砖面)中的所有线程分别运行下面的代码//把值复制到tile_static实例nums,所以同一个nums会被赋值2*2=4次nums[idx.local[1]][idx.local[0]] = sample[idx.global];//等待tile(砖面)中的所有线程,运行完上面这段代码idx.barrier.wait();//现在nums中的2*2=4个元素已经有有效值了//tile(砖面)中的所有线程再次分别运行下面的代码//计算平均值,int sum = nums[0][0] + nums[0][1] + nums[1][0] + nums[1][1];//把计算结果复制到array_view对象中.average[idx.global] = sum / 4;});//打印运算结果for (int i = 0; i < 4; i++) {for (int j = 0; j < 6; j++) {std::cout << average(i, j) << " ";}std::cout << "\n";}// Output:// 3 3 8 8 3 3// 3 3 8 8 3 3// 5 5 2 2 4 4// 5 5 2 2 4 4}


   使用平铺的好处是,从tile_static变量存取数据要比从全局空间(array和array_view对象)要快。为了从平铺中得到性能优势,我们的算法必须把计算域拆分为tile(砖面)然后把数据放到tile_static变量中加快数据存取速度。

   注意不要使用类似下面的代码来累加tile(砖面)中的数据,

tile_static  float  total;

total  +=  matrix[t_idx];‘

原因[1]total的初始值是不确定的,所以第二句代码的运算没有意义。

原因[2]由于tile(砖面)中的多根线程竞争同一个title_static变量,计算结果会不确定。

 

内存屏障(MemoryFences)

         在restrict(amp)限定中,有两种内存必须要同步:

全局内存:array或array_view实例

tile_static内存:tile(砖面)内存

         内存屏障确保两种内存的线程同步,要调用内存屏障可以使用下面三种方法:

         tile_barrier::wait(或tile_barrier::wait_with_all_memory_fence)方法: 建立全局内存和tile_static内存的屏障。

   tile_barrier::wait_with_global_memory_fence方法 : 仅建立全局内存的屏障

   tile_barrier::wait_with_tile_static_memory_fence 方法 :仅建立tile_static内存的屏障

   调用特定类型的屏障(fence)可以提高你应用的性能,在下面的例子中 tile_barrier::wait_with_tile_static_memory_fence 方法的调用代替tile_barrier::wait方法的调用提高了应用的性能。

// 使用tile_static内存屏障parallel_for_each(matrix.extent.tile<SAMPLESIZE, SAMPLESIZE>(),     [=, &averages](tiled_index<SAMPLESIZE, SAMPLESIZE> t_idx) restrict(amp){    // 把数据从全局内存中复制到title_static内存中.    tile_static floattileValues[SAMPLESIZE][SAMPLESIZE];   tileValues[t_idx.local[0]][t_idx.local[1]] = matrix[t_idx];     // 等待title_static内存中的数据复制完毕   t_idx.barrier.wait_with_tile_static_memory_fence();     // 如果你移除if语句,代码会把被tile(砖面)中的所有线程调用,这样每个    // tile(砖面)中的元素都会被分配一个同样的平均值.    if (t_idx.local[0] == 0&& t_idx.local[1] == 0) {        for (int trow = 0; trow <SAMPLESIZE; trow++) {            for (int tcol = 0; tcol< SAMPLESIZE; tcol++) {               averages(t_idx.tile[0],t_idx.tile[1]) += tileValues[trow][tcol];            }        }       averages(t_idx.tile[0],t_idx.tile[1]) /= (float) (SAMPLESIZE *SAMPLESIZE);    }});


         restrict(amp)修饰的代码段是在加速器(GPU)上运行的,默认在里面的代码段,下断点不会Break(进入),在[Solution Explorer]窗口中点击项目名称,快捷键[Alt]+[Enter],打开当前项目属性页,[Configuration Properties]->[Debugging]->[Debugger Type]默认为“Auto”,改为“GPU Only”就可以Debug当前项目加速器(GPU)上运行的代码了。

         据官网介绍,无符号整数的处理速度要比带符号整数快,所以尽量用无符号整数吧。

参考资料

《Using Tiles》

http://msdn.microsoft.com/en-us/library/vstudio/hh873135.aspx

《使用平铺》

http://msdn.microsoft.com/zh-cn/library/vstudio/hh873135.aspx


4 0