# op-plugin_1

**Repository Path**: aahehe/op-plugin_1

## Basic Information

- **Project Name**: op-plugin_1
- **Description**: OpPlugin of Ascend Extension for PyTorch
- **Primary Language**: Unknown
- **License**: Not specified
- **Default Branch**: 7.0.0
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 766
- **Created**: 2024-07-12
- **Last Updated**: 2025-07-01

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# OpPlugin

## 简介

本项目开发了Ascend Extension for Pytorch（torch_npu）算子插件，为使用PyTorch框架的开发者提供便捷的NPU算子库调用能力。
OpPlugin算子插件的编译、使用依赖昇腾Ascend Extension for PyTorch。因此，在编译OpPlugin之前，需要了解、安装昇腾PyTorch。使用手册可参考昇腾社区[Ascend Extension for Pytorch](https://gitee.com/ascend/pytorch/blob/master/README.zh.md)。

## 编译、安装OpPlugin

### 发布包安装
暂未正式发布

### 源码安装

**安装依赖**

安装时需要安装系统依赖及官方PyTorch框架，建议使用torch_npu提供的docker镜像进行编译，依赖安装和镜像使用指导可参考[Ascend Extension for Pytorch](https://gitee.com/ascend/pytorch/blob/master/README.zh.md#%E4%BD%BF%E7%94%A8%E6%BA%90%E7%A0%81%E7%BC%96%E8%AF%91%E5%AE%89%E8%A3%85pytorch%E6%A1%86%E6%9E%B6)。

1. 配置CANN环境变量脚本。

   ```
   source <CANN软件安装目录>/<CANN软件路径>/set_env.sh
   ```

   环境变量脚本的默认路径一般为：/usr/local/Ascend/ascend-toolkit/set_env.sh，其中ascend-toolkit路径取决于安装的CANN软件名称。

2. 编译生成插件的二进制安装包。

   下载对应OpPlugin版本分支代码，进入插件根目录，以v2.1为例
   ```
   git clone https://gitee.com/ascend/op-plugin.git -b 7.0.0
   cd op-plugin
   ```
   执行编译构建，当前支持torch_npu 2.1/2.3/2.4/2.5版本
   ```
   bash ci/build.sh --python=3.8 --pytorch=v2.1.0-7.0.0
   ```
   请注意，编译时gcc版本遵循如下约束：
   arm架构下推荐使用gcc 10.2, x86架构下推荐使用gcc 9.3.1; 2.6及之后版本的编译推荐使用gcc 11.2.1

   | 参数      | 取值范围                                           | 说明                  | 缺省值    | 备注                                |
   |---------|------------------------------------------------|---------------------|--------|-----------------------------------|
   | pytorch | v2.1.0-7.0.0, v2.3.1-7.0.0, v2.4.0-7.0.0, v2.5.1-7.0.0 | 指定编译过程中使用的pytorch版本 | master | 需要显示指定pytorch版本，避免使用缺省值master            |
   | python  | 3.8, 3.9, 3.10, 3.11                           | 指定编译过程中使用的python版本  | 3.8    | 编译基于pytorch2.5及后续版本时，不支持python3.8 |


   >编译过程中，会在插件根目录新建build文件夹，并下载torch_npu对应版本的源码，协同编译。 若build/pytorch目录存在，则编译op-plugin时，不再重复下载torch_npu源码。如需下载所依赖的最新torch_npu源码，删除build/pytorch目录即可。

3. 完成编译后，安装dist目录下生成的插件torch\_npu包，如果使用非root用户安装，需要在命令后加**--user**。

   ```
   pip3 install --upgrade dist/torch_npu-{torch_npu_version}-{pytohon_version}-{arch}.whl
   # 实际执行时需要根据生成的whl名称替换上述whl包名称，其中{version}表示编译的torch_npu版本，{python_version} 为所使用的 Python 版本，{arch} 则代表目标架构
   # 典型的whl包名类似：torch_npu-2.1.0.post13+gitb32f3-cp38-cp38-manylinux_2_17_aarch64.manylinux2014_aarch64.whl
   ```

## 版本配套表
op-plugin仓旨在为**torch_npu**提供运行所需要的算子适配文件，两个仓的对应关系如下：
| op-plugin分支 | 对应Ascend Extension for PyTorch版本 |
| ------------- | :----------------------------------: |
| master        |     主线版本，如v2.1.0等     |
| 7.0.0        |     7.0.0版本，如v2.1.0-7.0.0等       |
| 6.0.0        |     6.0.0版本，如v2.1.0-6.0.0等       |
| 6.0.rc3       |   6.0.rc3版本，如v2.1.0-6.0.rc3等    |
| 6.0.rc2       |   6.0.rc2版本，如v2.1.0-6.0.rc2等    |
| 6.0.rc1       |   6.0.rc1版本，如v2.1.0-6.0.rc1等    |
| 5.0.0         |     5.0.0版本，如v2.1.0-5.0.0等      |
| 5.0.rc3       |   5.0.rc3版本，如v2.1.0-5.0.rc3等    |

## 支持的型号
- Atlas 训练系列产品
- Atlas A2 训练系列产品

## 生命周期
op-plugin仓依赖**torch_npu**运行，生命周期请参考**torch_npu**中的[PyTorch版本维护策略](https://gitee.com/ascend/pytorch/blob/v2.1.0-7.0.0/README.zh.md#pytorch%E7%89%88%E6%9C%AC%E7%BB%B4%E6%8A%A4%E7%AD%96%E7%95%A5)。

## 安全声明

[OpPlugin安全声明](SECURITYNOTE.md)