CANN/ops-fft构建参数说明

build.sh 参数说明

【免费下载链接】ops-fftops-fft 是 CANN （Compute Architecture for Neural Networks）算子库中提供 FFT 类计算的基础算子库，采用模块化设计，支持灵活的算子开发和管理。项目地址: https://gitcode.com/cann/ops-fft

简介

build.sh 是 ops-fft 项目的构建脚本，位于项目根目录下。该脚本通过配置不同参数实现多种功能，包括编译算子、运行测试、生成安装包等。

使用方法

1. 配置环境变量

在使用 build.sh 之前，需要先配置 CANN 环境变量：

# 默认路径安装 source /usr/local/Ascend/cann/set_env.sh # 验证环境变量 echo $ASCEND_HOME_PATH

2. 构建命令格式

./build.sh [OPTIONS]

参数说明

build.sh 支持多种功能，可通过--help参数查看所有选项：

./build.sh --help

支持的 SoC 型号

使用示例

基本编译

# 编译所有算子（默认 8 线程） ./build.sh # 编译指定算子 ./build.sh --ops=rfft1_d # 编译多个算子 ./build.sh --ops=rfft1_d,irfft1_d # 使用 16 线程编译 ./build.sh -j16

编译并测试

# 编译所有算子并运行测试 ./build.sh --run # 编译指定算子并运行测试 ./build.sh --ops=rfft1_d --run # 指定测试超时时间（600 秒） ./build.sh --run --test-timeout=600

打包

# 编译所有算子并打包（默认 SoC: Ascend950） ./build.sh --pkg # 编译指定算子并打包 ./build.sh --ops=rfft1_d --pkg # 为指定 SoC 打包 ./build.sh --soc=Ascend950 --pkg # 大小写不敏感 ./build.sh --soc=ascend950 --pkg

组合使用

# 编译 rfft1_d 算子、运行测试、使用 16 线程 ./build.sh --ops=rfft1_d --run -j16 # 编译所有算子、打包、指定 SoC ./build.sh --pkg --soc=Ascend950 -j16

行为说明

输出说明

编译输出

编译成功后，生成的文件位于build/目录：

build/ ├── libops_fft.so # 动态库 ├── tests/ │ └── all_ops_test # 测试可执行文件 └── ...

打包输出

打包成功后，会在build/目录生成 .run 文件：

build_out/cann-{soc}-ops-fft_{version}_linux-{arch}.run

例如：

• cann-950-ops-fft_9.0.0_linux-x86_64.run
• cann-950-ops-fft_9.0.0_linux-aarch64.run

注意事项

1. 环境变量要求：必须设置ASCEND_HOME_PATH环境变量，否则脚本会报错退出。

2. 线程数限制：如果指定的线程数超过 CPU 核心数，脚本会自动调整为 CPU 核心数。

3. 测试超时：默认测试超时时间为 300 秒，可根据实际情况调整。

4. SoC 大小写：--soc参数支持大小写不敏感输入，脚本会自动标准化为首字母大写、其余小写的格式。

5. 算子验证：如果使用--ops指定了不存在的算子，脚本会列出所有可用的算子并报错退出。

相关文档

• 环境部署
• 算子调用
• 算子开发

【免费下载链接】ops-fftops-fft 是 CANN （Compute Architecture for Neural Networks）算子库中提供 FFT 类计算的基础算子库，采用模块化设计，支持灵活的算子开发和管理。项目地址: https://gitcode.com/cann/ops-fft