FINN Development in a VS Code Dev Container (on Windows using WSL 2)

[fpjentzsch]

Hi,

I've been working with FINN in Visual Studio Code on Windows via a WSL 2 Ubuntu distribution and Docker Desktop for a while. This works fine, but the disconnect between writing code in the WSL workspace but executing it within the FINN Docker container (via run-docker.sh) is not ideal.

To work directly within the container, I created the devcontainer.json file I'm sharing here. It allows VS Code to spin up the FINN Docker environment as a dev container, in which your development workspace resides. This has the following benefits:

• Full IntelliSense (e.g., Pylance) with up to date dependencies
• Pytest integration
• Python debugger integration
• Ability to launch multiple terminals or Jupyter notebooks directly in VS Code
• It's even easier to work with local forks of dependencies (simply comment-in the respective lines in the devcontainer.json)

Setup

Besides some VS Code configuration via settings.json and launch.json files, you'll need to append the following code to Dockerfile.finn for this to work:

ARG USERNAME=finn-user
ARG USER_UID=1000
ARG USER_GID=$USER_UID
RUN groupadd --gid $USER_GID $USERNAME \
    && useradd --uid $USER_UID --gid $USER_GID -m $USERNAME \
    && echo $USERNAME ALL=\(root\) NOPASSWD:ALL > /etc/sudoers.d/$USERNAME \
    && chmod 0440 /etc/sudoers.d/$USERNAME
USER $USERNAME

Other considerations:

• Your workspace (where finn/ is cloned) should not be = $HOME, otherwise the container will mount your entire $HOME.
• Make sure to set all environment variables required by FINN, especially those that have no default value defined in devcontainer.json.

Feedback

Please share your ideas for improving this setup here. I'll try to maintain this post accordingly.
Note that this is not an officially supported installation of FINN.

[fpjentzsch]

I also tried to experiment with a custom BuildKit frontend, for example this one to apply the patch to Dockerfile.finn by extending instead of modifying it.

Unfortunately, custom frontends like this are broken in dev containers until microsoft/vscode-remote-release#6848 is addressed.

[kalahel]

Hello,

I have a quick question, how do you manage FINN call to vitis_hls since it is a windows install of Vitis ?

Personally when launching notebooks/end2end_example/bnn-pynq/cnv_end2end_example.py I have AssertionError: vitis_hls not found in PATH even though I have given the xilinx installation path.
Also my /mnt/c/Tools/Xilinx/Vitis_HLS/2024.1/bin/vitis_hls cannot run since there is no linux folder installation.

That what I get if I try to run it :

ERROR: Could not find 64-bit executable.
ERROR: /mnt/c/Tools/Xilinx/Vitis_HLS/2024.1/bin/unwrapped/lnx64.o/vitis_hls does not exist

[fpjentzsch]

Hi,

I actually install Vitis/Vivado inside my WSL 2 distribution so FINN can use it. If you have WSLg enabled, you can simply use the GUI-based Linux web installer. Otherwise you can execute the installer via command line.

[kalahel]

Thanks, that's very clever !