CANN/atvoss API接口列表
ATVOSS接口列表【免费下载链接】atvossATVOSSAscend C Templates for Vector Operator Subroutines是一套基于Ascend C开发的Vector算子库致力于为昇腾硬件上的Vector类融合算子提供极简、高效、高性能、高拓展的编程方式。项目地址: https://gitcode.com/cann/atvoss根据ATVOSS架构分层和模板归属不同提供给用户可用接口列表如表1所示表1ATVOSS接口列表分类模板接口说明所属头文件入参构造NAArgumentsBuilder::inputOutput用户输入输出构造方法/include/utils/arguments/arguments.hArgumentsBuilder::attr用户属性构造方法ArgumentsBuilderImpl::build用户参数对象生成器Device层ElewiseDeviceAdapterdevice适配层对象构造函数/include/elewise/device/device_adapter.hDeviceAdapter::Rundevice适配层主运行接口Kernel层ElewiseKernelBuilderkernel层对象构造函数/include/elewise/kernel/builder.hDefaultKernelPolicy默认的kernel层静态policyDefaultKernelConfig默认的kernel层schedule配置参数的数据结构DefaultKernelSchedule默认的kernel层schedule调度策略/include/elewise/kernel/schedule.hBaseKernelSchedule::MakeScheduleConfigkernel层schedule基类的生成scheduleCfg配置信息方法BaseKernelSchedule::Runkernel层schedule基类的执行调度策略方法Block层ElewiseBlockBuilderblock层对象构造函数/include/elewise/block/builder.hDefaultBlockPolicy默认的block层静态policyDefaultBlockConfig默认的block层schedule配置参数的数据结构DefaultBlockSchedule默认的block层schedule调度策略/include/elewise/block/schedule.hBaseBlockSchedule::MakeScheduleConfigblock层schedule基类的生成scheduleCfg配置信息方法BaseBlockSchedule::Runblock层schedule基类的执行调度策略方法Compute层NACompute用户表达Compute运算逻辑关系的静态配置NAPlaceHolder在Compute表达中用户定义参数对象的函数方法/include/expression/expr_template.hPlaceHolderTmpLike在Compute表达中用户定义临时对象的函数方法ParamUsage指定参数的数据流向UnaryOp一元运算符BinaryOp二元运算符TernaryOp三元运算符为了方便用户表达compute运算逻辑提供给用户运算符接口列表如表2所示表2Operator接口列表操作符名称操作符说明加法运算-减法运算*乘法运算/除法运算Exp以自然常数e为底的指数运算Power幂运算Sqrt开平方运算Cast数据类型转换运算Abs绝对值运算【免费下载链接】atvossATVOSSAscend C Templates for Vector Operator Subroutines是一套基于Ascend C开发的Vector算子库致力于为昇腾硬件上的Vector类融合算子提供极简、高效、高性能、高拓展的编程方式。项目地址: https://gitcode.com/cann/atvoss创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考