nngenインストール

NNgen: A Fully-Customizable Hardware Synthesis Compiler for Deep Neural Network

をインストールしたので、メモ。

環境:macOS Mojave 10.14.6

$ python3 --version
Python 3.7.4

nngenインストール

READMEを見ながらポチポチと。

Icarus Verilog

$ brew install icarus-verilog

pyverilog, veriloggen, onnx

$ pip3 install jinja2 pyverilog veriloggen numpy onnx
:
Installing collected packages: pyverilog, veriloggen, onnx
Successfully installed onnx-1.6.0 pyverilog-1.2.0 veriloggen-1.8.1

pytestやPyTorchインストール

私は、先にPyTorchはインストールしてしまっていたのでスキップされて必要なものだけ。

$ pip3 install pytest pytest-pythonpath torch torchvision
:
Installing collected packages: packaging, pluggy, py, pytest, pytest-pythonpath
Successfully installed packaging-19.2 pluggy-0.13.1 py-1.8.0 pytest-5.3.2 pytest-pythonpath-0.7.3

テスト実行

$ cd nngen/tests
$ python3 -m pytest .
================================================= test session starts ==================================================
platform darwin -- Python 3.7.4, pytest-5.3.2, py-1.8.0, pluggy-0.13.1
rootdir: /Users/users/nngen, inifile: pytest.ini
plugins: pythonpath-0.7.3
collected 284 items

matrix_add/test_matrix_add_int16_par2.py .                                                                       [  0%]
matrix_add/test_matrix_add_int16_wrap_par2.py .                                                                  [  0%]
matrix_add/test_matrix_add_int2_par16.py .                                                                       [  1%]
matrix_add/test_matrix_add_int32.py .                                                                            [  1%]

ってことで、まだ終わってなくてファンがファンファンいってる。