diff --git a/.gitignore b/.gitignore index 8b3166a44070a4575aac86c445c4504b594cda08..d7ee7e014a0c175a8a88060f2aa320efeb501ddc 100644 --- a/.gitignore +++ b/.gitignore @@ -81,3 +81,6 @@ MANIFEST # SSH key dir mounted into Docker /ssh_keys/ + +# PYNQ board files +/board_files/ diff --git a/docker/Dockerfile.finn_ci b/docker/Dockerfile.finn_ci index 7d5772d9f5118d1f1238dd14a6b57a1b4fd5004d..0d122133a6446cb77160c9447e16ff13d4d4b9c5 100644 --- a/docker/Dockerfile.finn_ci +++ b/docker/Dockerfile.finn_ci @@ -37,7 +37,7 @@ RUN apt-get update RUN apt-get -y upgrade RUN apt-get install -y build-essential libglib2.0-0 libsm6 libxext6 libxrender-dev RUN apt-get install -y verilator zsh -RUN apt-get -y install sshpass +RUN apt-get -y install sshpass wget unzip RUN echo "StrictHostKeyChecking no" >> /etc/ssh/ssh_config # cloning dependency repos diff --git a/docker/Dockerfile.finn_dev b/docker/Dockerfile.finn_dev index 8c1502eb4a1941061bd58e6f9a18106f98f259e2..f8e15f34fb4da3dc4ee353a29d26866b68879144 100644 --- a/docker/Dockerfile.finn_dev +++ b/docker/Dockerfile.finn_dev @@ -43,7 +43,7 @@ RUN apt-get update RUN apt-get -y upgrade RUN apt-get install -y build-essential libglib2.0-0 libsm6 libxext6 libxrender-dev RUN apt-get install -y verilator nano zsh rsync -RUN apt-get -y install sshpass +RUN apt-get -y install sshpass wget unzip RUN echo "StrictHostKeyChecking no" >> /etc/ssh/ssh_config COPY requirements.txt . diff --git a/docker/finn_entrypoint.sh b/docker/finn_entrypoint.sh index ee75089c657e4fad1e4a455ac7bd5fe4976e5d4c..72751817383dbdb441970e5816247cfa7760ef5b 100644 --- a/docker/finn_entrypoint.sh +++ b/docker/finn_entrypoint.sh @@ -57,4 +57,19 @@ if [ ! -z "$VITIS_PATH" ];then export XILINX_VITIS=$VITIS_PATH source $VITIS_PATH/settings64.sh fi + +# download PYNQ board files if not already there +if [ ! -d "/workspace/finn/board_files" ]; then + gecho "Downloading PYNQ board files for Vivado" + wget -q https://github.com/cathalmccabe/pynq-z1_board_files/raw/master/pynq-z1.zip + wget -q https://d2m32eurp10079.cloudfront.net/Download/pynq-z2.zip + unzip -q pynq-z1.zip + unzip -q pynq-z2.zip + mkdir /workspace/finn/board_files + mv pynq-z1/ board_files/ + mv pynq-z2/ board_files/ + rm pynq-z1.zip + rm pynq-z2.zip +fi + exec "$@"