CANN ops-fft算子开发快速入门

算子开发快速入门基于ops-fft仓【免费下载链接】ops-fftops-fft 是 CANN Compute Architecture for Neural Networks算子库中提供 FFT 类计算的基础算子库采用模块化设计支持灵活的算子开发和管理。项目地址: https://gitcode.com/cann/ops-fft本指南旨在帮助你快速上手基于CANN和ops-fft算子仓的使用最简化地完成环境安装、编译部署及算子运行。一、环境安装1. 有环境场景Docker安装Docker安装环境以Atlas A2产品910B为例。前提条件Docker环境宿主机已安装Docker引擎版本1.11.2及以上。驱动与固件宿主机已安装昇腾NPU的驱动与固件Ascend HDK 24.1.0版本以上。安装指导详见《CANN 软件安装指南》。注意使用npu-smi info查看对应的驱动与固件版本。下载镜像拉取已预集成CANN软件包及ops-fft所需依赖的镜像。操作步骤以root用户登录宿主机。执行拉取命令请根据你的宿主机架构选择ARM架构docker pull --platformarm64 swr.cn-south-1.myhuaweicloud.com/ascendhub/cann:8.5.0-910b-ubuntu22.04-py3.10-opsX86架构docker pull --platformamd64 swr.cn-south-1.myhuaweicloud.com/ascendhub/cann:8.5.0-910b-ubuntu22.04-py3.10-ops注意正常网速下镜像下载时间约为5-10分钟。Docker运行请根据以下命令运行dockerdocker run --name cann_container --device /dev/davinci0 --device /dev/davinci_manager --device /dev/devmm_svm --device /dev/hisi_hdc -v /usr/local/dcmi:/usr/local/dcmi -v /usr/local/bin/npu-smi:/usr/local/bin/npu-smi -v /usr/local/Ascend/driver/lib64/:/usr/local/Ascend/driver/lib64/ -v /usr/local/Ascend/driver/version.info:/usr/local/Ascend/driver/version.info -v /etc/ascend_install.info:/etc/ascend_install.info -it swr.cn-south-1.myhuaweicloud.com/ascendhub/cann:8.5.0-910b-ubuntu22.04-py3.10-ops bash检查环境进入容器后验证环境和驱动是否正常。检查NPU设备npu-smi info检查CANN安装cat /usr/local/Ascend/ascend-toolkit/latest/opp/version.info2. 无环境场景WebIDE开发建设中对于无环境的用户提供WebIDE开发方式目前本方式正在建设中。二、编译部署1. 拉取ops-fft仓库代码git clone https://gitcode.com/cann/ops-fft.git cd ops-fft2. 编译ops-fft算子包bash build.sh --pkg若提示如下信息说明编译成功。Self-extractable archive cann-ops-fft-1.0.0-linux-*.run successfully created. Build package success: build/cann-ops-fft-1.0.0-linux-*.run编译成功后run包存放于项目根目录的build目录下。3. 安装ops-fft算子包./build/cann-ops-fft-*linux*.runops-fft安装在${ASCEND_HOME_PATH}/cann路径中${ASCEND_HOME_PATH}表示CANN软件安装目录。4. 配置环境变量source ${ASCEND_HOME_PATH}/cann/set_env.bash5. 运行测试bash build.sh --run三、更多帮助CANN 开发文档Ascend C API 参考【免费下载链接】ops-fftops-fft 是 CANN Compute Architecture for Neural Networks算子库中提供 FFT 类计算的基础算子库采用模块化设计支持灵活的算子开发和管理。项目地址: https://gitcode.com/cann/ops-fft创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考