如何使用c++和CUDA的Cooperative Groups进行GPU线程协作? (网格级同步)

夏浩酱_4187

夏浩酱_4187

2026-01-22

141人浏览

原创

cooperative_groups网格级同步的本质是启动必须全部参与且原子性完成的跨SM kernel,并用cg::grid_group::synchronize()实现全局屏障;需满足计算能力≥6.0、TCC模式(Tesla卡)、空流三条件,且仅在该kernel内生效。

如何使用c++和cuda的cooperative groups进行gpu线程协作? (网格级同步)

什么是 cooperative_groups 网格级同步的本质

它不是“让所有 GPU 线程等一个栅栏”,而是允许你启动一个跨多个 SM 的、**必须全部参与且原子性完成**的 kernel,并在其中用 cg::grid_group 调用 synchronize()。前提是:kernel 必须用 cudaLaunchCooperativeKernel 启动,且设备支持(cudaDeviceGetAttribute(&attr, cudaDevAttrCooperativeLaunch, dev) 返回非零),常见于 Tesla A100 / V100 / H100,消费卡(如 RTX 4090)默认禁用或不支持。

启动 cooperative kernel 的三个硬性条件

缺一不可,否则 cudaLaunchCooperativeKernel 直接返回 cudaErrorNotSupported 或 cudaErrorInvalidValue:

  • GPU 计算能力 ≥ 6.0(Pascal),但实际需看驱动与模式:Tesla 卡需在 TCC 模式下运行;Windows WDDM 下一律不支持
  • Host 端调用前必须设置流为 NULL:cudaStream_t stream = 0,不能传自定义流
  • Kernel 中若使用 cg::grid_group,必须在 kernel 入口第一行声明并获取,例如:
    __global__ void my_kernel() {
        namespace cg = cooperative_groups;
        cg::grid_group grid = cg::this_grid();
        // 后续才能调用 grid.synchronize()
    }

cg::grid_group::synchronize() 的行为边界

它只对当前 cooperative kernel 内所有线程生效,不阻塞 host,也不影响其他 kernel。但它会强制等待:所有 SM 上该 kernel 的所有 block 都执行到该语句,且所有 warp 都到达后才继续。注意以下陷阱:

  • 不能在 if (threadIdx.x == 0) 这类 divergent 分支里调用——会导致部分 warp 永远不执行,kernel hang 住
  • 不能和 __syncthreads() 混用在同个 block 内做嵌套同步——无定义行为,常见死锁
  • 每个 block 内只能调用一次 grid.synchronize()(多次调用不报错但无意义,且可能触发驱动异常)
  • 同步开销显著:A100 上单次耗时约 5–10 μs,比 __syncthreads() 高两个数量级

典型协作模式:全局归约 + 分布式 barrier

适合需要所有 block 协同完成阶段性计算的场景,比如多 block 共同构建哈希表、分阶段排序、或分布式 SGD 的梯度聚合。下面是最小可运行结构:

deepstream-dev
deepstream-dev

使用 Python pyservicemaker API 进行 NVIDIA DeepStream SDK 开发。在构建视频分析流水线、基于 GStreamer

下载
__global__ void global_reduce_kernel(float* data, int n) {
    namespace cg = cooperative_groups;
    cg::grid_group grid = cg::this_grid();
    extern __shared__ float sdata[];
<pre class="brush:php;toolbar:false;">int tid = threadIdx.x;
int bid = blockIdx.x;
int offset = bid * blockDim.x;

// Step 1: block 内规约到 shared memory
sdata[tid] = (offset + tid  0; s >>= 1) {
    if (tid <p>}</p>

真正难的从来不是写这十几行代码,而是确认你的运行环境是否满足 cooperative launch 的全部约束——尤其是驱动模式、设备属性、流参数这三个点,漏查一个,kernel 就静默失败或直接 crash。

C++免费学习笔记(深入):立即使用
在学习笔记中,你将探索 C++ 的入门与实战技巧!

相关文章

c++速学教程(入门到精通)
c++速学教程(入门到精通)

c++怎么学习?c++怎么入门?c++在哪学?c++怎么学才快?不用担心,这里为大家提供了c++速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

windows c++ stream c++ cuda

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
什么是分布式
什么是分布式

分布式是一种计算和数据处理的方式,将计算任务或数据分散到多个计算机或节点中进行处理。本专题为大家提供分布式相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.11

2213

5

分布式和微服务的区别
分布式和微服务的区别

分布式和微服务的区别在定义和概念、设计思想、粒度和复杂性、服务边界和自治性、技术栈和部署方式等。本专题为大家提供分布式和微服务相关的文章、下载、课程内容,供大家免费下载体验。

2023.10.07

3714

5

c语言中null和NULL的区别
c语言中null和NULL的区别

c语言中null和NULL的区别是:null是C语言中的一个宏定义,通常用来表示一个空指针,可以用于初始化指针变量,或者在条件语句中判断指针是否为空;NULL是C语言中的一个预定义常量,通常用来表示一个空值,用于表示一个空的指针、空的指针数组或者空的结构体指针。

2023.09.22

569

3

java中null的用法
java中null的用法

在Java中,null表示一个引用类型的变量不指向任何对象。可以将null赋值给任何引用类型的变量,包括类、接口、数组、字符串等。想了解更多null的相关内容,可以阅读本专题下面的文章。

2024.03.01

1698

6

线程和进程的区别
线程和进程的区别

线程和进程的区别:线程是进程的一部分,用于实现并发和并行操作,而线程共享进程的资源,通信更方便快捷,切换开销较小。本专题为大家提供线程和进程区别相关的各种文章、以及下载和课程。

2023.08.10

4878

6

windows查看端口占用情况
windows查看端口占用情况

Windows端口可以认为是计算机与外界通讯交流的出入口。逻辑意义上的端口一般是指TCP/IP协议中的端口,端口号的范围从0到65535,比如用于浏览网页服务的80端口,用于FTP服务的21端口等等。怎么查看windows端口占用情况呢?php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

2023.07.26

3499

3

查看端口占用情况windows
查看端口占用情况windows

端口占用是指与端口关联的软件占用端口而使得其他应用程序无法使用这些端口,端口占用问题是计算机系统编程领域的一个常见问题,端口占用的根本原因可能是操作系统的一些错误,服务器也可能会出现端口占用问题。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.07.27

3038

6

windows照片无法显示
windows照片无法显示

当我们尝试打开一张图片时,可能会出现一个错误提示,提示说"Windows照片查看器无法显示此图片,因为计算机上的可用内存不足",本专题为大家提供windows照片无法显示相关的文章,帮助大家解决该问题。

2023.08.01

1548

3

windows查看端口被占用的情况
windows查看端口被占用的情况

windows查看端口被占用的情况的方法:1、使用Windows自带的资源监视器;2、使用命令提示符查看端口信息;3、使用任务管理器查看占用端口的进程。本专题为大家提供windows查看端口被占用的情况的相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.02

2759

5

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Python uv官方安装与入门文档
Python uv官方安装与入门文档

共0课时 | 0人学习

Buffalo框架快速入门指南
Buffalo框架快速入门指南

共0课时 | 0人学习

Conan 2 安装指南
Conan 2 安装指南

共0课时 | 0人学习