讲师中心 微信公众号
AI工具推荐 视频效率加速

C++ 怎么在 GPU 上实现高效的并行前缀和(Scan / Prefix Sum)算法

夏婷吖_4273

夏婷吖_4273

发布时间:2026-10-21 11:08:22

|

255人浏览过

|

来源于php中文网

原创

因为std::inclusive_scan是CPU端顺序/线程并行实现,不生成GPU可执行kernel,也无法利用warp-level原语(如__shfl_down_sync),故无法直接在GPU上运行。

c++ 怎么在 gpu 上实现高效的并行前缀和(scan / prefix sum)算法

为什么不能直接用 std::inclusive_scan 上 GPU?

因为 std::inclusive_scan 是 CPU 端顺序/线程并行实现,不生成 GPU 可执行的 kernel,也不能利用 warp-level 原语(比如 __shfl_down_sync)。直接丢数组到 CUDA kernel 里手写串行前缀和,会彻底失去并行性——每个线程等前一个算完,O(n) 时间,毫无意义。

GPU 上高效 Scan 必须分层:先做 block 内快速 scan(用 shared memory + warp shuffle),再对各 block 的末尾元素做全局 scan(reduce-then-scan),最后用 block scan 结果修正本块输出。这是经典两阶段法(Hillis-Steele 或 Blelloch 变种),CUDA Toolkit 自带的 thrust::exclusive_scan 就是这么干的。

用 thrust::device_vector + thrust::exclusive_scan 最简可行路径

90% 场景下,别自己重写。Thrust 已针对不同数据规模、GPU 架构(Turing/Ampere)自动选择最优策略(比如小数组用单 block,大数组切分+多级 reduce)。

  • 输入必须是 device-side 连续内存:thrust::device_vector<int></int>,不是 std::vector 或裸指针(否则触发隐式 host-device 拷贝,性能归零)
  • 注意 inclusive vs exclusive:如果需要 [a₀, a₀+a₁, a₀+a₁+a₂],用 thrust::inclusive_scan;若要 [0, a₀, a₀+a₁],才用 exclusive_scan
  • 自定义二元操作符(如 max-scan)需确保满足结合律,且 CUDA 支持该 functor 的 device 编译(加 __device__ __host__)
thrust::device_vector<float> d_in = {...};
thrust::device_vector<float> d_out(d_in.size());
thrust::inclusive_scan(d_in.begin(), d_in.end(), d_out.begin());

手动写 block-level scan 时,__shfl_down_sync 的坑在哪?

Warp shuffle 是 block 内 scan 的核心加速点,但极易出错:

C++14
C++14

C++14 对 C++11 的修正与增强版本,适合旧系统维护和较老工具链兼容。

下载

立即学习“C++免费学习笔记(深入)”;

  • 必须用 mask(如 0xffffffff),不能传 0 —— 否则在部分 warp 活跃线程数
  • 偏移量 delta 必须是 2 的幂(1,2,4,8,16),且 threadIdx.x % 32 >= delta 的线程才参与 shuffle,否则读到旧值或越界
  • shared memory 的 bank conflict:若按线程号顺序存 sdata[tid],当 tid=0,16,32… 时会撞同一 bank,拖慢 5–10 倍;改用 sdata[tid ^ 16] 或 padding 可缓解

典型错误现象:cuda-memcheck 报 invalid address,或结果每 32 个元素出现重复/跳变——基本就是 shuffle mask 或 bank conflict 搞错了。

大规模数据(>1GB)Scan 的吞吐瓶颈常不在算法,而在内存访问模式

即便用了最优两阶段 scan,实测发现瓶颈常是 global memory 的 coalescing 效率:

  • block scan 阶段,每个 warp 读 d_in[base + tid], d_in[base + tid + 1], ... 是连续的,没问题
  • 但 global scan 阶段,各 block 只取末尾一个元素(如 d_in[block_id * N - 1]),这些地址跨距极大,变成随机访存,带宽利用率掉到 30% 以下
  • 解决办法:预处理阶段把所有 block 尾部元素 gather 到连续数组(用 thrust::gather + stride iterator),再 scan 这个紧凑数组,最后 scatter 回去

这个细节在论文里常被忽略,但实测对 2GB 浮点数组能提升 1.7× 总耗时——尤其在 A100/H100 上,global memory 带宽吃不满比算力不足更常见。

热门AI工具

更多
豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

二狗PPT
二狗PPT Hot

一款AI演示文稿工具,主要用于专为中式职场打造的AI PPT生成工具,适合需要提升相关任务效率的用户。

切问学术

切问学术是一款AI论文写作工具,复旦大学NLP团队推出的AI学术智能体。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

Laper
Laper Hot

Laper是专为编剧、导演和制片人推出的 AI 原生剧本创作工具。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

SkildArt
SkildArt Hot

SkildArt是一款AI文本写作工具,一站式 AI 视觉创作平台。

讯飞绘文

讯飞绘文是一款由科大讯飞推出的一站式 AIGC 内容运营平台。

立刻MV
立刻MV Hot

立刻MV是一款AI文本写作工具,AI 音乐视频(MV)创作工具。

相关专题

更多
c++和c语言的区别有哪些
c++和c语言的区别有哪些

c++和c语言的区别:1、面向对象编程(OOP)支持不同;2、新增特性不同;3、标准库不同;4、编译方式不同;5、命名空间不同等等。想了解更多c++的相关内容,可以阅读本专题下面的文章。

2248

2024.03.14

c++和python学习顺序推荐
c++和python学习顺序推荐

一般建议先学习C++,再学习Python,因为这样可以逐步从较为底层的编程语言向更高级的语言过渡。想了解更多python的相关内容,可以阅读本专题下面的文章。

1019

2024.03.14

python和c++学习性价比分析
python和c++学习性价比分析

Python易于学习,广泛应用于Web开发、数据科学和人工智能等领域,但性能较低。C语言性能高,适用于对性能要求较高的场景,如游戏开发和系统编程,但学习曲线陡峭,错误处理复杂。想了解更多python的相关内容,可以阅读本专题下面的文章。

447

2024.03.14

c语言和c++一样吗
c语言和c++一样吗

c语言和c++是两种不同的编程语言,虽然有相似之处,但存在显著差异。c语言专注于过程式编程和系统级开发,以简洁、高效著称。c++作为c语言的超集,引入了面向对象编程,增强了代码组织和管理能力,但学习曲线也更陡峭。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

327

2024.03.14

c语言和c++先学哪个好
c语言和c++先学哪个好

初学者选择学习c语言还是c++语言,需要根据个人学习目标、背景以及编程兴趣和预期应用方向来决定。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

426

2024.03.14

c语言和c++的区别和联系
c语言和c++的区别和联系

c语言和c++是计算机科学领域应用广泛的编程语言。虽然它们有着相似的基础,但它们在语言类型、语法功能和内存管理方面存在着显著差异。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

580

2024.03.14

c++软件中文更改教程
c++软件中文更改教程

对于 ide,可通过打开设置,找到语言设置,选择中文,并保存更改。对于非 ide 应用程序,可查找设置或选项,选择语言设置,更改为中文,并保存更改。想了解更多c++的相关内容,可以阅读本专题下面的文章。

1409

2024.03.21

python和java和c++学习性价比分析
python和java和c++学习性价比分析

Python以其易学性、丰富的库和活跃的社区而著称,适合数据科学、人工智能和Web开发。Java以其跨平台性、企业级应用开发和Android应用开发而闻名。C++以其底层控制能力、高效性能和游戏开发而著称。选择哪种语言取决于个人兴趣、职业方向和特定需求。想了解更多python和java和c++的相关内容,可以阅读本专题下面的文章。

1217

2024.03.22

Gradle插件开发入门教程
Gradle插件开发入门教程

本专题介绍Gradle插件开发的基本流程,涵盖插件类编写、自定义任务注册、扩展参数设计和插件应用方法,帮助你理解插件如何组织构建逻辑,减少多个项目之间的重复配置。

0

2026.10.22

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
C++ Core Guidelines
C++ Core Guidelines

共0课时 | 0人学习

C++ Reference
C++ Reference

共0课时 | 0人学习

C++ 官方标准说明
C++ 官方标准说明

共0课时 | 0人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn