# tk_cannops

**Repository Path**: opentk/tk_cannops

## Basic Information

- **Project Name**: tk_cannops
- **Description**: 基于昇腾NPU平台开发的适用于电力巡检场景的算子和模型加速库
- **Primary Language**: C++
- **License**: Apache-2.0
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 3
- **Forks**: 3
- **Created**: 2025-08-22
- **Last Updated**: 2026-03-05

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

## tk_cannops

### 介绍

tk_cannops 是一个基于昇腾 NPU 平台开发的高性能算子和模型加速库，专为电力巡检场景设计。它提供了一系列优化后的算子和加速接口，支持 PyTorch 框架，旨在提高模型在昇腾 NPU 上的推理和训练性能。

### 快速上手

1. 确保你的环境已安装昇腾 NPU 的相关依赖和驱动。
2. 编译并安装算子：`bash ./tk_cannops/build.sh`
3. 运行测试案例：`bash ./test/run.sh`

### 目录结构

```
.
├── tk_cannops                  # 高性能算子库
│  ├── api                      # 加速库API适配层
│  ├── ops                      # 算子实现
│  │  ├── op_host               # host侧实现
│  │  └── op_kernel             # kernel侧实现
├── model_examples              # 场景模型示例
│  └── Qwenvl2.5
├── docs                        # 文档
├── LICENSE                     # 开源协议
├── README.md                   # 项目说明
└── tests                       # 测试文件
```

### 算子清单

本项目提供多种高性能算子，具体清单将在后续补充，敬请期待。

### 模型清单

目前支持的场景模型示例包括 Qwenvl2.5，更多模型将在后续版本中加入。

### 运行模型示例

模型示例位于 `model_examples` 目录中，按照目录内的 README 指南进行操作。

### 参与贡献

欢迎参与贡献！以下是简单的步骤：

1. Fork 本仓库。
2. 新建 Feat_xxx 分支。
3. 提交你的代码。
4. 创建 Pull Request。

### 依赖

- 昇腾NPU固件
- 昇腾NPU驱动
- CANN


### 开源协议

本项目遵循开源协议，请查看 `LICENSE` 文件了解详细许可协议。

### 联系我们

如有问题，请提交 issue 或联系项目维护者。