__cvta_ubuf_to_generic

产品支持情况

产品	是否支持
Atlas 350 加速卡	√
Atlas A3 训练系列产品/Atlas A3 推理系列产品	x
Atlas A2 训练系列产品/Atlas A2 推理系列产品	x
Atlas 200I/500 A2 推理产品	x
Atlas 推理系列产品AI Core	x
Atlas 推理系列产品Vector Core	x
Atlas 训练系列产品	x

功能说明

将Unified Buffer内存空间的地址值转换为对应的指针并返回。

函数原型

void* __cvta_ubuf_to_generic(size_t rawbits)

参数说明

表1 参数说明
参数名	输入/输出	描述
rawbits	输入	源操作数。

返回值说明

指向Unified Buffer内存空间的指针。

约束说明

SIMD与SIMT混合编程场景不支持使用该接口。

需要包含的头文件

使用该接口需要包含"simt_api/device_functions.h"头文件。

#include "simt_api/device_functions.h"

调用示例

SIMT编程场景：

__global__ __launch_bounds__(1024) void kernel__cvta_ubuf_to_generic(uint32_t* dst, uint32_t* src)
{
    __ubuf__ uint32_t data_ptr[1024];
    int idx = threadIdx.x + blockIdx.x * blockDim.x;
    data_ptr[threadIdx.x] = src[idx];
    size_t addr = __cvta_generic_to_ubuf(data_ptr + threadIdx.x);
    void* ptr = __cvta_ubuf_to_generic(addr);
    dst[idx] = *(uint32_t*)ptr;
}

父主题： 地址空间转换函数