
DotEx 算子【免费下载链接】ops-blas本项目是CANN提供的高性能线性代数计算以及轻量化GEMM调用算子库。项目地址: https://gitcode.com/cann/ops-blas算子概述扩展精度向量点积算子实现两个向量的点积运算支持混合精度计算。内部统一以 FP32 精度累加最终结果转换为输入类型。数学表达式result Σ(x[i * incx] * y[i * incy]), i 0 ... n-1包含以下接口接口名功能简述aclblasDotEx扩展精度向量点积算子执行接口aclblasDotEx产品支持情况Ascend 950PR / Ascend 950DT支持接口说明aclblasStatus_t aclblasDotEx( aclblasHandle_t handle, int n, const void *x, aclDataType xType, int incx, const void *y, aclDataType yType, int incy, void *result, aclDataType resultType, aclDataType executionType)参数输入/输出说明handle输入算子句柄n输入向量元素个数x输入向量 x 的设备地址xType输入x 的数据类型incx输入x 的存储步长y输入向量 y 的设备地址yType输入y 的数据类型incy输入y 的存储步长result输出点积结果的设备地址resultType输入result 的数据类型executionType输入中间计算精度固定为 ACL_FLOAT返回值值说明ACLBLAS_STATUS_SUCCESS成功ACLBLAS_STATUS_HANDLE_IS_NULLPTRhandle 为空ACLBLAS_STATUS_INVALID_VALUE参数非法ACLBLAS_STATUS_NOT_SUPPORTED不支持的类型组合ACLBLAS_STATUS_ALLOC_FAILED内存分配失败ACLBLAS_STATUS_EXECUTION_FAILED执行失败支持的数据类型x/y 类型result 类型execution 类型FP32FP32FP32FP16FP16FP32BF16BF16FP32注意事项x 和 y 的数据类型必须相同result 类型必须与 x/y 类型一致executionType 必须为 ACL_FLOAT编译与运行bash build.sh --run --opsdotex --socascend950【免费下载链接】ops-blas本项目是CANN提供的高性能线性代数计算以及轻量化GEMM调用算子库。项目地址: https://gitcode.com/cann/ops-blas创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考