Metadata-Version: 2.2
Name: pto-kernels
Version: 0.1.1
Summary: PTO kernels
Author: HUAWEI Computing Systems Lab
Maintainer: Anastasios Zouzias, Gioele Gottardo, Aleksandros Sobczyk, Jiawei Zhuang
Classifier: Programming Language :: Python :: 3 :: Only
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Requires-Python: >=3.10
Requires-Dist: pybind11==3.0.1
Requires-Dist: numpy==1.26.4
Requires-Dist: scipy==1.13.1
Requires-Dist: decorator==5.1.1
Requires-Dist: psutil==6.1.0
Requires-Dist: PyYAML==6.0.2
Requires-Dist: attrs==24.2.0
Provides-Extra: torch
Requires-Dist: torch-npu==2.8.0.post2; extra == "torch"
Description-Content-Type: text/markdown

# pto-isa kernels

Ascend NPU kernels using [pto-isa](https://gitcode.com/cann/pto-isa/). Parallel Tile Operation (PTO) is a virtual instruction set architecture designed by Ascend CANN, focusing on tile-level operations.

## Build

```bash
source /usr/local/Ascend/ascend-toolkit/set_env.sh
pip3 install -r requirements.txt
make build_wheel
```

The above commands will generate a wheel (i.e., `pto_kernels-0.1.0-*.whl`) that is pip installable.

### Installing

```bash
pip install --force-reinstall pto_isa_kernels-*.whl
```

```bash
make test
```
