AllocScalar

【免费下载链接】opbase本项目是CANN算子库的基础框架库，为算子提供公共依赖文件和基础调度能力。项目地址: https://gitcode.com/cann/opbase

功能说明

申请一个aclScalar对象，并对其赋值。通过多个重载函数，用于支持多种数据类型。

函数原型

aclScalar *AllocScalar(const void *data, op::DataType dataType)

aclScalar *AllocScalar(float value)

aclScalar *AllocScalar(double value)

aclScalar *AllocScalar(fp16_t value)

aclScalar *AllocScalar(bfloat16 value)

aclScalar *AllocScalar(int32_t value)

aclScalar *AllocScalar(int64_t value)

aclScalar *AllocScalar(int16_t value)

aclScalar *AllocScalar(int8_t value)

aclScalar *AllocScalar(uint32_t value)

aclScalar *AllocScalar(uint64_t value)

aclScalar *AllocScalar(uint16_t value)

aclScalar *AllocScalar(uint8_t value)

aclScalar *AllocScalar(bool value)

参数说明

返回值说明

申请到的aclScalar对象，申请失败返回nullptr。

约束说明

入参指针不能为空。

调用示例

// 初始化一个值为5，数据类型为int64的aclScalar对象 void Func(aclOpExecutor *executor) { int64_t val = 5; aclScalar *scalar = executor->AllocScalar(val); scalar = executor->AllocScalar(&val, DT_INT64); }

【免费下载链接】opbase本项目是CANN算子库的基础框架库，为算子提供公共依赖文件和基础调度能力。项目地址: https://gitcode.com/cann/opbase