OpenCV CUDA模块设备层-----GPU上执行线程安全的 “原子取最大值” 操作函数

  • 操作系统:ubuntu22.04
  • OpenCV版本:OpenCV4.9
  • IDE:Visual Studio Code
  • 编程语言:C++11

算法描述

这是一个 OpenCV 的 CUDA 模块(cv::cudev) 中封装的原子操作函数,用于在 GPU 上执行线程安全的 "原子取最大值" 操作。

将 *address 和 val 进行比较,如果 val > *address,就将 *address = val。

函数原型

cpp 复制代码
__device__ __forceinline__ int cv::cudev::atomicMax(int* address, int val)

参数

参数名 类型 含义
address int* 一个指向设备内存中整型变量的指针。这是你要进行原子最大值比较的目标地址。
val int 要和目标地址值比较的整数值。如果 val > *address,则更新为 val。

返回值

返回修改前 *address 的原始值。

代码示例

cpp 复制代码
#include <cuda_runtime.h>
#include <cstdio>

__global__ void kernel(int* max_value) {
    int tid = threadIdx.x + blockIdx.x * blockDim.x;

    // 使用 CUDA 内建的 atomicMax
    int old = atomicMax(max_value, tid);

    printf("Thread %d: old = %d, new = %s\n",
           tid, old, (tid > old ? "updated" : "not updated"));
}

int main() {
    int h_max = 0;
    int* d_max;

    // 分配设备内存
    cudaMalloc(&d_max, sizeof(int));
    if (!d_max) {
        printf("Failed to allocate device memory!\n");
        return -1;
    }

    // 初始化设备内存
    cudaMemcpy(d_max, &h_max, sizeof(int), cudaMemcpyHostToDevice);

    // 启动核函数
    kernel<<<1, 10>>>(d_max);

    // 显式同步设备
    cudaDeviceSynchronize();  // 等待核函数执行完成

    // 拷贝结果回主机
    cudaMemcpy(&h_max, d_max, sizeof(int), cudaMemcpyDeviceToHost);

    printf("Final max value: %d\n", h_max);  // 应该输出 9

    // 清理资源
    cudaFree(d_max);
    return 0;
}

运行结果

bash 复制代码
Thread 0: old = 0, new = not updated
Thread 1: old = 0, new = updated
Thread 2: old = 1, new = updated
Thread 3: old = 2, new = updated
Thread 4: old = 3, new = updated
Thread 5: old = 4, new = updated
Thread 6: old = 5, new = updated
Thread 7: old = 6, new = updated
Thread 8: old = 7, new = updated
Thread 9: old = 8, new = updated
Final max value: 9
相关推荐
数说故事1 天前
数说故事 | 2025年运动相机数据报告,深挖主流品牌运营策略及行业趋势
大数据·人工智能·aigc·数说故事
明月(Alioo)1 天前
机器学习入门,支持向量机
人工智能·机器学习·支持向量机
Ronin-Lotus1 天前
深度学习篇---pytorch数据集
人工智能·pytorch·深度学习
云雾J视界1 天前
人月神话今犹在:从布鲁克斯法则到阿里云AI代码生成
人工智能·项目管理·ai编程·人月神话·人机月
算家计算1 天前
DeepSeek被曝年底推出AI智能体,下一代人机交互时代要来了?
人工智能·agent·deepseek
HenrySmale1 天前
01 神经网络简介
人工智能·深度学习·神经网络
爱补鱼的猫猫1 天前
pytorch可视化工具(训练评估:Tensorboard、swanlab)
人工智能·pytorch·python
算家计算1 天前
腾讯最新开源HunyuanVideo-Foley本地部署教程:端到端TV2A框架,REPA策略+MMDiT架构,重新定义视频音效新SOTA!
人工智能·开源
格林威1 天前
Linux使用-Linux系统管理
linux·运维·服务器·深度学习·ubuntu·计算机视觉
moonsheeper1 天前
NLP技术爬取
人工智能·自然语言处理