CANN/Ascend C内存预留API

ReserveLocalMemory【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit功能说明该函数用于在Unified Buffer中预留指定大小的内存空间。调用该接口后使用GetCoreMemSize可以获取实际可用的剩余Unified Buffer空间大小。函数原型void ReserveLocalMemory(ReservedSize size)参数说明参数名输入/输出说明ReservedSize输入需要预留的空间大小。enum class ReservedSize { RESERVED_SIZE_8K, // 预留8 * 1024B空间 RESERVED_SIZE_16K, // 预留16 * 1024B空间 RESERVED_SIZE_32K, // 预留32 * 1024B空间 };返回值说明无约束说明多次调用该函数时仅保留最后一次调用的结果。调用示例ge::graphStatus TilingXXX(gert::TilingContext* context) { auto ascendcPlatform platform_ascendc::PlatformAscendC(context-GetPlatformInfo()); uint64_t ub_size, l1_size; // 预留8KB的Unified Buffer内存空间 ascendcPlatform.ReserveLocalMemory(platform_ascendc::ReservedSize::RESERVED_SIZE_8K); // 获取Unified Buffer和L1的实际可用内存大小 ascendcPlatform.GetCoreMemSize(platform_ascendc::CoreMemType::UB, ub_size); ascendcPlatform.GetCoreMemSize(platform_ascendc::CoreMemType::L1, l1_size); // ... return ret; }完整样例可参考与数学库高阶API配合使用的样例。【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考