CANN/atvoss ArgumentsBuilder输入输出方法文档

ArgumentsBuilder::inputOutput【免费下载链接】atvossATVOSSAscend C Templates for Vector Operator Subroutines是一套基于Ascend C开发的Vector算子库致力于为昇腾硬件上的Vector类融合算子提供极简、高效、高性能、高拓展的编程方式。项目地址: https://gitcode.com/cann/atvoss功能说明用户侧输入输出对象构造方法支持用户传入输入/输出tensor以及host侧的一些属性参数函数参数中的参数列表和Compute表达中的参数列表在时序上完全对应两者配合使用。所属头文件链接/include/utils/arguments/arguments.h函数原型struct ArgumentsBuilder { templatetypename... InitialInputOutput constexpr auto inputOutput(InitialInputOutput... inputOutput) const }参数说明参数名称参数类型输入/输出数据类型参数说明默认值InitialInputOutput模板参数输入NA可变参数列表类型其中每个参数类型必须为Tensor或非指针类的scalar根据用户传入的参数列表实例化NAinputOutput函数形参输入InitialInputOutput用户传入的参数列表NA返回值说明返回值数据类型返回值说明ArgumentsBuilderImpl返回参数构建器对象约束说明NA使用示例template typename InputDtype, typename OutputDtype struct AddSubConfig { struct AddSubCompute { template template typename class Tensor __host_aicore__ constexpr auto Compute() const { auto in1 Atvoss::PlaceHolder1, TensorInputDtype, Atvoss::ParamUsage::IN(); auto in2 Atvoss::PlaceHolder2, TensorInputDtype, Atvoss::ParamUsage::IN(); auto in3 Atvoss::PlaceHolder3, InputDtype, Atvoss::ParamUsage::IN(); auto out Atvoss::PlaceHolder4, TensorOutputDtype, Atvoss::ParamUsage::OUT(); return (out in1 in2 - in3); }; }; using ArchTag Atvoss::Arch::DAV_3510; using BlockOp Atvoss::Ele::BlockBuilderAddSubCompute, ArchTag; using KernelOp Atvoss::Ele::KernelBuilderBlockOp; using DeviceOp Atvoss::DeviceAdapterKernelOp; }; template typename InputDtype, typename OutputDtype static void Run() { /* ACL init and stream create */ ... Atvoss::TensorInputDtype in1(deviceIn1, {{3, 4, 0, 0, 0, 0, 0, 0}}, 2); Atvoss::TensorInputDtype in2(deviceIn2, {{3, 4, 0, 0, 0, 0, 0, 0}}, 2); InputDtype in3 5.0; Atvoss::TensorOutputDtype out(deviceOut, {{3, 4, 0, 0, 0, 0, 0, 0}}, 2); // 使用示例 auto arguments Atvoss::ArgumentsBuilder{}.inputOutput(in1, in2, in3, out).build(); // 使用示例 using DeviceOp typename AddSubConfigInputDtype, OutputDtype::DeviceOp; DeviceOp deviceOp; deviceOp.Run(arguments, stream); } int main(int argc, char const* argv[]) { Runfloat, float(); return 0; }【免费下载链接】atvossATVOSSAscend C Templates for Vector Operator Subroutines是一套基于Ascend C开发的Vector算子库致力于为昇腾硬件上的Vector类融合算子提供极简、高效、高性能、高拓展的编程方式。项目地址: https://gitcode.com/cann/atvoss创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考