CANN/ops-solver批量复数矩阵求逆算子

CgetriBatched算子实现【免费下载链接】ops-solver本项目是CANN提供的高级数值求解算子库实现矩阵分解、求逆、特征值求解等功能在NPU上的加速计算。项目地址: https://gitcode.com/cann/ops-solver概述Solver CgetriBatched算子实现。支持的产品Atlas A3 训练系列产品/Atlas A3 推理系列产品Atlas A2 训练系列产品/Atlas A2 推理系列产品目录结构介绍├── cgetri_batched │ ├── CMakeLists.txt // 编译工程文件 │ ├── README.md // 说明文档 │ └── cgetri_batched_test.cpp // 算子调用样例算子描述算子功能CgetriBatched算子计算批量复数矩阵的逆矩阵对应的数学表达式为 $$A^{-1}A I$$ 其中$A$为$n \times n$阶非奇异复数方阵$I$为$n$阶单位矩阵。算子规格算子类型(OpType)CgetriBatched算子输入nameshapedata typeDescriptionformatn[1]INT64单个矩阵A的行数\A[batchSize, n, n]COMPLEX64公式中的矩阵A行主序NDlda[1]INT64A左右相邻元素间的内存地址偏移量当前约束为n\lda_inv[1]INT64输出的逆矩阵的左右相邻元素间的内存地址偏移量当前约束为n\info[batchSize, 1]INT32每个batch矩阵的求逆结果信息NDbatchSize[1]INT64复数矩阵求逆中的矩阵数量\算子输出Ainv[batchSize, n, n]COMPLEX64输出的逆矩阵ND核函数名cgetri_batched_kernel算子约束lda、lda_inv、info参数在当前版本实际未启用。入参n大于等于32且小于等于256。入参batchSize小于等于3000。当n小于32时请使用CmatinvBatched算子。调用实现使用内核调用符调用核函数。编译运行在本样例根目录下执行如下步骤编译并执行算子。配置环境变量请根据当前环境上CANN开发套件包的安装方式选择对应配置环境变量的命令。默认路径root用户安装CANN软件包source /usr/local/Ascend/cann/set_env.sh默认路径非root用户安装CANN软件包source $HOME/Ascend/cann/set_env.sh指定路径install_path安装CANN软件包source ${install_path}/cann/set_env.sh样例执行bash build.sh --opscgetri_batched --run # --ops算子名 --run可选参数执行测试样例执行结果如下说明精度对比成功。[Success] Case accuracy is verification passed.【免费下载链接】ops-solver本项目是CANN提供的高级数值求解算子库实现矩阵分解、求逆、特征值求解等功能在NPU上的加速计算。项目地址: https://gitcode.com/cann/ops-solver创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考