爨爨爨好

  博客园  :: 首页  :: 新随笔  :: 联系 :: 订阅 订阅  :: 管理

▶ 协作组,要求 cuda ≥ 9.0,一个简单的例子见 http://www.cnblogs.com/cuancuancuanhao/p/7881093.html

● 灵活调节需要进行通讯的线程组合(不一定是线程块或是线程束)的尺寸,在更多粒度上进行线程协作。

● 协作组功能支持 CUDA 的各种并行模式,包括生产者 - 消费者并行(producer-consumer parallelism),机会并行(opportunistic parallelism),全网个同步(global synchronization)。

● 构成要素:① 参与协作的线程组合(即协作组整体)的数据类型;② 从 CUDA lauch API 中创建协作组(intrinsic groups?)的操作;③ 将现有协作组划分为新的协作组的操作;④ 协作组内的栅栏同步函数;⑤ 检查组内属性和执行组内特定命令的操作(如线程表决函数)。

● 块内协作组(Intra-block Group)使用方法。

 1 # include <cooperative_groups.h>        // 使用的头文件
 2 
 3 using namespace cooperative_groups;     // 命名空间
 4 
 5 thread_block g = this_thread_block();   // 将当前线程块打包为一个协作组,命名为 g
 6 
 7 thread_group gTile = tiled_partition(g, SIZE);
 8 // 将之前的协作组分割成大小为 SIZE 的协作组(SIZE 可以取 1,2,4,8,16,32),但组内不能使用线程束表决函数和统筹函数
 9 
10 thread_block_tile<SIZE> gTile = tiled_partition<SIZE>(g);
11 // 同样的分割函数,使用模板函数,编译时处理,比函数 tiled_partition() 更高效,且组内可以使用线程束表决函数和统筹函数
12 
13 // 协作组的一些方法
14 void sync();                            // 协作组同步(协作组内的线程栅栏同步)
15 unsigned size();                        // 获得协作组的大小(线程个数)
16 unsigned thread_rank();                 // 获得当前线程在协作组内的编号
17 bool is_valid();                        // 协作组是否有效(符合 API 约束)
18 dim3 group_index();                     // 指出当前线程块在线程格中的编号
19 dim3 thread_index();                    // 指出当前线程在线程块中的编号
20 
21 // 协作组内也可以使用的表决函数和统筹函数(成员函数)
22 int shfl();
23 int shfl_down();
24 int shfl_up();
25 int shfl_xor();
26 int any();
27 int all();
28 int ballot();
29 int match_any();
30 int match_all();

● 线程束发生分支的时候设备将会串行执行每个分支,在同道中保持活跃的所有线程称为合并的,协作组有能力发现并为合并的线程创建一个组。

1 coalesced_group active = coalesced_threads();// 在分支中,将当前活跃的线程创建为一个协作组

● 发现模式。两个示例代码段等价,但没看懂在干什么。

 1 {
 2     unsigned int writemask = __activemask();
 3     unsigned int total = __popc(writemask);
 4     unsigned int prefix = __popc(writemask & __lanemask_lt());
 5     // Find the lowest-numbered active lane
 6     int elected_lane = __ffs(writemask) - 1;
 7     int base_offset = 0;
 8     if (prefix == 0)
 9         base_offset = atomicAdd(p, total);
10     base_offset = __shfl_sync(writemask, base_offset, elected_lane);
11     int thread_offset = prefix + base_offset;
12     return thread_offset;
13 }
14 {
15     cg::coalesced_group g = cg::coalesced_threads();
16     int prev;
17     if (g.thread_rank() == 0)
18         prev = atomicAdd(p, g.size());
19     prev = g.thread_rank() + g.shfl(prev, 0);
20     return prev;
21 }

 ● 线程格同步,需要额外的一些步骤。

 1 // 通过 CUDA Driver API 的函数 cuDeviceGetAttribute() 来检查设备是否支持 cooperative launch 属性
 2 int pi = 0;
 3 cuDevice dev;
 4 cuDeviceGet(&dev, 0)
 5 cuDeviceGetAttribute(&pi, CU_DEVICE_ATTRIBUTE_COOPERATIVE_LAUNCH, dev);// 如果支持,则 pi 被置 1
 6 
 7 // 使用函数 cudaLaunchCooperativeKernel() 或 CUDA Driver API 中的几种调用方法来启动内核,不能使用 <<< >>>
 8 cudaLaunchCooperativeKernel(const T *func, dim3 gridDim, dim3 blockDim, void **args, size_t sharedMem = 0, cudaStream_t stream = 0);
 9 
10 // 建议精心优化线程格尺寸和线程块尺寸(下面两例分别是使用最大线程块数和自动优化线程块数)
11 {
12     cudaDeviceProp deviceProp;
13     cudaGetDeviceProperties(&deviceProp, dev);
14     cudaLaunchCooperativeKernel((void*)my_kernel, deviceProp.multiProcessorCount, numThreads, args);
15 }
16 {
17     cudaOccupancyMaxActiveBlocksPerMultiprocessor(&numBlocksPerSm, my_kernel, numThreads, 0));
18     cudaLaunchCooperativeKernel((void*)my_kernel, numBlocksPerSm, numThreads, args);
19 }
20 
21 // 使用函数 this_grid() 来获得当前线程格,以及使用线程格同步函数
22 grid_group grid = this_grid();
23 grid.sync();
24 
25 // 编译命令,打开 Relocatable Device Code(允许分离编译)
26 nvcc - arch = sm_61 - rdc = true mytestfile.cu - o mytest

 

● 多设备同步,需要额外的一些步骤。

 1 // 通过 CUDA Driver API 的函数 cuDeviceGetAttribute() 来检查设备是否支持 cooperative multi-device launch 属性
 2 int pi = 0;
 3 cuDevice dev;
 4 cuDeviceGet(&dev, 0)
 5 cuDeviceGetAttribute(&pi, CU_DEVICE_ATTRIBUTE_COOPERATIVE_MULTI_DEVICE_LAUNCH, dev);// 如果支持,则 pi 被置 1
 6 
 7 // 使用结构 CUDA_LAUNCH_PARAMS_st 来存储需要调用的内核的相关参数
 8 typedef struct CUDA_LAUNCH_PARAMS_st
 9 {
10     CUfunction function;
11     unsigned int gridDimX;
12     unsigned int gridDimY;
13     unsigned int gridDimZ;
14     unsigned int blockDimX;
15     unsigned int blockDimY;
16     unsigned int blockDimZ;
17     unsigned int sharedMemBytes;
18     CUstream hStream;
19     void **kernelParams;
20 }
21 CUDA_LAUNCH_PARAMS; 
22 
23 // 使用函数 cudaLaunchCooperativeKernelMultiDevice() 来启动内核,该函数允许主机线程创建一个跨设备的内核,以提供多设备同步功能
24 cudaLaunchCooperativeKernelMultiDevice(CUDA_LAUNCH_PARAMS *launchParamsList, unsigned int numDevices);
25 
26 // 使用函数 this_multi_grid() 来获得当前线程格,以及使用多设备同步函数
27 multi_grid_group multi_grid = this_multi_grid();
28 multi_grid.sync();
29 
30 // 编译命令,与线程格同步相同

■ 其他要点:

① 该 API 保证了操作的原子性,保证各主机线程在所有指定设备上独立的启动内核;不能将两个 launchParamsList 映射到同一个设备上

② 使用的所有设备必须具有相同的计算能力 major 和 minor 号;所有设备上使用的线程格尺寸、线程块尺寸和共享内存大小必须相同;通过该 API 启动的函数应该是相同的,API 内并没有内置相关检查。

③ 内核中使用的所有 __device__,__constant__,__managed__ 变量在各设备中相互独立,应该在启动内存钱分别初始化完成。

 

posted on 2017-12-11 00:24  爨爨爨好  阅读(936)  评论(0编辑  收藏  举报