A Fully-Customizable Hardware Synthesis Compiler for Deep Neural Network
Copyright 2017, Shinya Takamaeda-Yamazaki and Contributors
Apache License 2.0 (http://www.apache.org/licenses/LICENSE-2.0)
NNgenはディープニューラルネットワークのモデル特化ハードウェアアクセラレータを合成するオープンソースのコンパイラです。NNgenは入力モデル定義からDNNアクセラレータのVerilog HDLのソースコードとIPコアのパッケージを生成します。
生成されたハードウェアは、プロセッシングエンジン、オンチップメモリ、オンチップネットワーク、DMAコントローラ、制御回路といったすべてが含まれています。そのため、生成されたハードウェアは、処理を開始した後は、外部の回路やCPUなどからの追加の制御をまったく必要としません。
NNgenのバックエンドには、複数の記述パラダイムが利用できるオープンソースの高位合成コンパイラVeriloggenを用いています。そのため、新しいアルゴリズムやアプリケーションに応じて、NNgenの構成をカスタマイズすることができます。
NNgenプロジェクトは質問、バグ報告、新しい機能提案、GitHub上でのプルリクエストを常に歓迎します。
このプロジェクトの管理者として、コミュニティ管理を主導し、ソフトウェアの開発と普及を促進する役割を担います。
コミッタは、プロジェクトへの書き込み権限を付与されている個人です。コミッタとして貢献するためには、コミュニティ管理者の承認が必要です。貢献の分野は、コードの貢献とレビュー、ドキュメント、教育、そしてアウトリーチなど、あらゆる形態をとることができます。高品質で健全なプロジェクトにはコミッタが不可欠です。コミュニティは貢献者からの新しいコミッタを積極的に探しています。
レビュアーは、プロジェクトに積極的に貢献し、新しい貢献のコードレビューに積極的に参加する個人です。積極的な貢献者の中から査読者を決定します。 コミッタはレビュアーにレビューを明示的に依頼してください。 高品質のコードレビューは長期にわたる技術的債務を防ぎ、プロジェクトの成功に不可欠です。 プロジェクトへのプルリクエストは、マージするために少なくとも1人のレビュアーによってレビューされる必要があります。
GitHubのissue trackerにコメントを残してください。
プルリクエストを提供してくれた貢献者を確認するためには "CONTRIBUTORS.md"をチェックしてください。
NNgenは統合テストにpytestフレームワークを使います。 プルリクエストを送信するときは、pytestにテスト例を含めてください。 テストコードを書くには、 "tests"ディレクトリにある既存のテスト例を参照してください。
プルリクエスト対象のコードがすべてのテストをパスし、明白な問題がなければ、それはコミッタによってdevelopブランチにマージされます。
- Python3: 3.7.7 or later
- Apple Silicon上でのmacOS環境では、Python 3.10.6 (pyenvでインストール) を推奨します。
- Icarus Verilog: 10.1 or later
sudo apt install iverilog
- veriloggen: 2.3.0 or later
- numpy: 1.17 or later
- onnx: 1.9.0 or later
pip3 install veriloggen numpy onnx
python3 setup.py install
tests にいくつかのテストコードがあり、これらを実行するためには以下のソフトウェアが必要になります。
- pytest: 3.8.1 or later
- pytest-pythonpath: 0.7.3 or later
- PyTorch: 1.3.1 or later
- torchvision: 0.4.2 or later
pip3 install pytest pytest-pythonpath torch torchvision
高速なRTLシミュレーションにはVerilatorが必要となります。
- Verilator: 3.916 or later
sudo apt install verilator
- TeX Live: 2015 or later
- dvipng: 1.15 or later
sudo apt install texlive-science texlive-fonts-recommended texlive-fonts-extra dvipng
- Sphinx: 2.10 or later
- sphinx_rtd_theme : 0.4.3 or later
pip3 install sphinx sphinx_rtd_theme
Dockerfileを用いてNNgen用の環境を構築することができます。
cd docker
sudo docker build -t user/nngen .
sudo docker run --name nngen -i -t user/nngen /bin/bash
cd nngen/examples/mlp/
make
準備中です。英語版READMEをご参照下さい。
- PythonでVerilog HDLソースコードを構築するためのライブラリ
- PythonベースのVerilog HDLハードウェア設計処理ツールキット