diff --git a/docker/Dockerfile.finn_dev b/docker/Dockerfile.finn_dev index 4c8557b53b05bea67c96ad9edf523d532640ebfe..6e100a8792bee7b766caed107a4171e39bb4576d 100644 --- a/docker/Dockerfile.finn_dev +++ b/docker/Dockerfile.finn_dev @@ -73,6 +73,8 @@ USER $UNAME # cloning dependency repos (as user) # finn-base RUN git clone https://github.com/Xilinx/finn-base.git /workspace/finn-base +# finn-experimental +RUN git clone https://github.com/Xilinx/finn-experimental.git /workspace/finn-experimental # Brevitas RUN git clone https://github.com/Xilinx/brevitas.git /workspace/brevitas # CNPY diff --git a/docker/finn_entrypoint.sh b/docker/finn_entrypoint.sh index bd2338305ef24d98f582d09a140175a243c62c7e..225bf6fe367663d3355f4693091be43ef7efd753 100644 --- a/docker/finn_entrypoint.sh +++ b/docker/finn_entrypoint.sh @@ -13,6 +13,7 @@ gecho () { # checkout the correct dependency repo commits # the repos themselves are cloned in the Dockerfile FINN_BASE_COMMIT=8908c6a3f6674c4fa790954bd41c23ee5bf053df +FINN_EXP_COMMIT=6e64238bc13c493f4454893c635b4c13ed59e158 BREVITAS_COMMIT=aff49758ec445d77c75721c7de3091a2a1797ca8 CNPY_COMMIT=4e8810b1a8637695171ed346ce68f6984e585ef4 HLSLIB_COMMIT=2e49322d1bbc4969ca293843bda1f3f9c05456fc @@ -25,6 +26,11 @@ gecho "finn-base @ $FINN_BASE_COMMIT" git -C /workspace/finn-base pull --quiet git -C /workspace/finn-base checkout $FINN_BASE_COMMIT --quiet pip install --user -e /workspace/finn-base +# finn-experimental +gecho "finn-experimental @ $FINN_EXP_COMMIT" +git -C /workspace/finn-experimental pull --quiet +git -C /workspace/finn-experimental checkout $FINN_EXP_COMMIT --quiet +pip install --user -e /workspace/finn-experimental # Brevitas gecho "brevitas @ $BREVITAS_COMMIT" git -C /workspace/brevitas pull --quiet