# kunpeng-extension-for-pytorch

**Repository Path**: tiancheng-bao/kunpeng-extension-for-pytorch

## Basic Information

- **Project Name**: kunpeng-extension-for-pytorch
- **Description**: A package for extending the official Pytorch that can easily obtain performance on Kunpeng platform
- **Primary Language**: Unknown
- **License**: Not specified
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 2
- **Created**: 2025-07-31
- **Last Updated**: 2025-08-05

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# kpex

#### 介绍
kpex is a package for extending the official Pytorch that can easily obtain performance on Kunpeng platform

#### 安装教程

**一、环境设置**

1.安装pytorch

2.加载kutacc环境

```
module use /xxx/xxx/xxx/xxx/xxx/modulefiles
module load xxx/xxx/kutacc
```

3.安装依赖

```
pip install ninja=1.11.1.1 pybind11==2.11.1
```
**二、执行安装命令**
```
CFLAGS="-stdlib=libc++ -lc++abi" KPEX_BUILD_TYPE=release KUTACC_ROOT=/xxx/xxx/kutacc pip install --editable .
```
#### 卸载教程
```
pip uninstall kunpeng-pytorch-extension -y
```
#### 接口说明
接口名:
kpex_alphafold

接口描述:
通过替换模型中的算子来优化afphafold2的模型

接口参数:
|参数名|类型|描述|输入/输出|
|-----|----|-----|---------|
|model|模型类|alphafold2模型|输入|
|model_config|配置类|alphafold2模型的config|输入|
|new_model|模型类|优化后的alphafold2模型|输出|

#### 使用说明

```
model = kpex.tpp.alphafold.alphafold.kpex_alphafold(model, model_config)
```