Metadata-Version: 2.4
Name: torchao-npu
Version: 0.1.0.dev0
Summary: TorchAO quantization and training support for Ascend NPU
Home-page: https://gitcode.com/cann/torchtitan-npu
License: BSD-3-Clause
Classifier: Development Status :: 3 - Alpha
Classifier: Operating System :: POSIX :: Linux
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.12
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Requires-Python: >=3.12
Description-Content-Type: text/markdown
License-File: LICENSE
Requires-Dist: torch
Requires-Dist: torchao
Requires-Dist: torch-npu
Dynamic: classifier
Dynamic: description
Dynamic: description-content-type
Dynamic: home-page
Dynamic: license
Dynamic: license-file
Dynamic: requires-dist
Dynamic: requires-python
Dynamic: summary

<div align="center" markdown="1">

# torchao-npu

<h4>面向 Ascend NPU 的 torchao 低精度训练扩展</h4>

</div>

`torchao-npu` 为 `torchao` 提供 Ascend NPU 低精度训练相关的配置、量化原语和算子实现。
当前版本为开发预览版，接口和支持范围可能继续调整。

## 环境要求

- Python 3.12 或更高版本
- PyTorch
- 与 PyTorch 和 CANN 版本匹配的 `torch-npu`
- `torchao`
- 可用的 Ascend NPU 环境

安装 `torchao-npu` 前，应按照 Ascend 软件配套要求安装匹配版本的 PyTorch、`torch-npu` 和 CANN。

## 安装

从 PyPI 安装当前开发预览版：

```bash
python -m pip install torchao-npu==0.1.0.dev0
```

## 验证安装

```bash
python -c "import torchao_npu; print(torchao_npu.__file__)"
```

命令输出 `torchao_npu` 的安装路径即表示 Python 包可以正常导入。NPU 算子仍需在可用的 Ascend NPU
环境中验证。

## License

本项目使用 BSD 3-Clause License，具体请参见 [LICENSE](LICENSE)。
