文档
注册
评分
提单
论坛
小AI

GmAlloc

函数功能

进行核函数的CPU侧运行验证时,用于创建共享内存:在/tmp目录下创建一个共享文件,并返回该文件的映射指针。

函数原型

void *GmAlloc(size_t size)

参数说明

参数名

输入/输出

描述

size

输入

用户想要申请的共享内存大小

返回值

void* mem;

该指针表示该共享内存空间的首地址

支持的型号

Atlas 训练系列产品

Atlas推理系列产品AI Core

Atlas A2训练系列产品/Atlas 800I A2推理产品

Atlas 200/500 A2推理产品

约束说明

该接口在系统的/tmp目录下生成临时文件,故需要磁盘空间足够才可以正常生成共享内存。

调用示例

constexpr int32_t len = 8 * 32 * 1024 * 8;
half* x = (half*) GmAlloc(len*sizeof(half));
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词