在PGI的官方网站上获得示例代码:
http://www.pgroup.com/lit/samples/pgi_accelerator_examples.tar
我们的第一个例子从一个简单的程序开始。这个程序是把一个浮点向量送到GPU上,然后乘以2.再把结果返回。
整个程序是:
1 #include <stdio.h>
2 #include <stdlib.h>
3 #include <assert.h>
4 int main( int argc, char* argv[] )
5 {
6 int n; /* size of the vector */
7 float *a; /* the vector */
8 float *restrict r; /* the results */
9 float *e; /* expected results */
10 int i;
11 if( argc > 1 )
12 n = atoi( argv[1] );
13 else
14 n = 100000;
15 if( n <= 0 ) n = 100000;
16 a = (float*)malloc(n*sizeof(float));
17 r = (float*)malloc(n*sizeof(float));
18 e = (float*)malloc(n*sizeof(float));
19 /* initialize */
20 for( i = 0; i < n; ++i ) a[i] = (float)(i+1);
21 #pragma acc kernels loop
22 for( i = 0; i < n; ++i ) r[i] = a[i]*2.0f;
23 /* compute on the host to compare */
24 for( i = 0; i < n; ++i ) e[i] = a[i]*2.0f;
25 /* check the results */
26 for( i = 0; i < n; ++i )
27 assert( r[i] == e[i] );
28 printf( “%d iterations completed
”, n );
29 return 0;
30 }
请注意,在对指针r的声明中使用了 restrict 关键字; 我们很快就会知道为什么。
还请注意,定义浮点常量2.0f 而不是2.0 。在默认情况下,C语言的浮点常量是双精度。表达式a*2将会作为双精度计算,就好像(float)((double)a * 2.0)。为了避免这样,定义浮点常量,或者使用编译器选项-Mfcon,这样告诉编译器将浮点指针常量默认为通常的浮点来处理。
我们把进入GPU的循环之前放了一个kernals loop指令。这是告诉编译器去在查找循环的并行性,将数据传输到GPU上,在GPU上进行计算,然后将数据返回。对于这个程序,很简单。只需要输入下面的命令:
% pgcc -o c1.exe c1.c -acc -Minfo
也可以输入如下命令:
请注意-acc和-Minfo指令。-acc可以在编译器里启动OpenACC指令,默认情况下,PGI编译器认为目标加速区域在一个NVIDIA GPU上。我们会在后面的例子里演示其他的选项。
-Minfo指令就是从编译器上获得信息。我们将在我们的例子里解释这些信息代表的意思。当你在调试性能的时候,你会非常想了解这些信息的意思。
如果一切都安装正确,你就会从pgcc上获得以下信息:
1 main:
2 24, Generating copyout(r[:n])
3 Generating copyin(a[:n])
4 Generating compute capability 1.0 binary
5 Generating compute capability 2.0 binary
6 25, Loop is parallelizable
7 Accelerator kernel generated
8 25, #pragma acc loop gang, vector(128)
9 /* blockIdx.x threadIdx.x */
让我解释几个信息,首先是:
Generating copyout(r[:n])
就是告诉你在GPU上分配了数组r, 在循环执行后,把数据从GPU上复制回主机上。
Generating copyin(a[:n])
就是告诉你编译器认为数组a 为循环的输入,因此数组a的n个元素需要从CPU上复制到GPU上。
Loop is paralleizable
就是告诉你编译器分析循环体的参数后认为所有的迭代可以并行执行。我们增加了restrict关键词在指针r的声明,否则编译器不能确保a和r指向不同的mermory
Accelerator kernal generated
这是告诉你,编译器已经成功地把循环体转化成GPU的一个内核。这个内核是GPU自己的函数,由编译器产生,将会被程序调用,并在GPU上并行执行。
这样,你准备运行程序。假设你是在GPU设备上执行,只需要输入执行文件的名字,acc_c1.exe.如果你得到一个信息
libcuda.so not found, exiting
那么你可能是没有在它默认的位置安装CUDA驱动。你可能不得不设置环境变量LD_LIBRARY_PATH.
程序运行结束,你应该会看到一个结果:
100000 iterations completed
你如何知道哪些在GPU上执行?你可以设置环境变量ACC_NOTIFY为1.
csh: setenv ACC_NOTIFY 1
bash: export ACC_NOTIFY=1
然后重新运行程序。它会指出每次一个GPU内核执行的行数。在这个程序中,你会看到下面的结果:
launch kernel file=acc_c1.c function=main
line=25 device=0 grid=391 block=25
这是告诉你内核的文件名,函数,以及行数,CUDA grid和线程块的大小。
你可能不想给你所有的程序都设置这个环境变量,但在程序开发和测试的时候,这是个有用的办法。