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
相关推荐
Canace8 分钟前
我们是否需要AI知识库
人工智能·aigc·ai编程
en-route8 分钟前
从零开始学神经网络——前馈神经网络
人工智能·深度学习·神经网络
weixin_446260851 小时前
解锁安全新维度:Cybersecurity AI (CAI) 助力提升网络安全效率!
人工智能·安全·web安全
dream_home84071 小时前
构建生产级多模态数据集:视觉与视频模型(参照LLaVA-OneVision-Data和VideoChat2)
图像处理·人工智能·计算机视觉
AI优秘企业大脑1 小时前
音频库管理在数字媒体中的应用探索
大数据·人工智能
这儿有一堆花1 小时前
从图像到精准文字:基于PyTorch与CTC的端到端手写文本识别实战
人工智能·pytorch·python
聚客AI2 小时前
🌈从实验室到生产线:LLM工程师必须掌握的八大实战技能
人工智能·llm·agent
明天再做行么2 小时前
AI产品经理学习资料
人工智能·产品经理
cetcht88882 小时前
从 “有人值守” 到 “少人运维”:智能巡检机器人重塑配电室管理模式
大数据·运维·人工智能·机器人
阿正的梦工坊2 小时前
介绍 SWE-bench:语言模型能否解决真实世界的 GitHub 问题?
人工智能·语言模型·github