Skip to content

PAICookers/PAIlib

Repository files navigation

Library of PAICORE

PyPI - Python Version PyPI - Version CodeFactor Grade pre-commit.ci status License

paicorelib 是 PAICORE 系列的 Python 支持库,提供坐标、路由、寄存器模型、帧生成、帧解析和载荷打包等基础能力。

English

安装

pip install paicorelib

需要 Python 3.10 或更新版本。

主要能力

  • 坐标、硬件参数、计算核和神经元定义
  • 寄存器参数模型
  • 路由工具
  • 全局信号方向编码工具
  • 帧生成工具
  • PAICORE 2.5 配置帧流解析工具
  • Dense、CSC、BF16、FP32 等权重与载荷打包工具

文件组织

src/paicorelib/
├─ __init__.py          # 包级公共导出入口
├─ coordinate.py        # 坐标、偏移和坐标工具
├─ hw_defs.py           # 硬件参数和常量定义
├─ core_defs*.py        # 计算核寄存器枚举和限制
├─ core_model*.py       # 计算核寄存器参数模型
├─ neuron_defs*.py      # 神经元寄存器枚举和限制
├─ neuron_model*.py     # 神经元寄存器参数模型
├─ routing_defs.py      # PAICORE 多播路由基础类型
├─ routing_hexa.py      # PAICORE 2.5 NoC / AER 路由和全局信号编码工具
├─ float_codec.py       # BF16/FP32 数值载体和载荷编码工具
└─ framelib/
    ├─ frame_defs.py    # 帧格式和字段定义
    ├─ frames.py        # 帧对象定义
    ├─ frame_gen*.py    # 帧生成工具
    ├─ parser_v2.py     # PAICORE 2.5 配置帧流解析
    ├─ types.py         # 帧数组和载荷类型
    └─ utils.py         # 帧载荷辅助工具

文档

开发

uv sync --group dev
uv run pytest -q

About

Library of PAICORE.

Topics

Resources

License

Stars

4 stars

Watchers

1 watching

Forks

Contributors

Languages