Dockerfile

# Copyright (C) 2023 Advanced Micro Devices, Inc. All rights reserved.
# SPDX-License-Identifier: MIT

FROM ubuntu:24.04

ARG BUILD_TEMPDIR
ARG LIC_MAC=00:00:00:00:00:00
ENV LIC_MAC=${LIC_MAC}

ARG USER_ID
ARG GROUP_ID
ARG GROUP_NAME

SHELL ["/bin/bash", "-c"]

RUN apt-get update
RUN apt-get install -y sudo adduser perl

RUN deluser --remove-home $(getent passwd | awk -F: -v uid="$USER_ID" '$3 == uid {print $1}') || echo "Ignoring error, as uid:${USER_ID} does not exist"
RUN groupadd -g ${GROUP_ID} ${GROUP_NAME} || echo "Ignoring error, as ${GROUP_NAME} already exists"
RUN useradd -u $USER_ID -g $GROUP_ID -ms /bin/bash riallto
RUN adduser riallto sudo
RUN echo 'riallto ALL=(ALL) NOPASSWD: ALL' > /etc/sudoers.d/riallto
USER riallto
ENV HOME /home/riallto
WORKDIR /home/riallto

RUN sudo chmod a+rwx /root

ARG DEBIAN_FRONTEND=noninteractive

# Generate locales
RUN sudo apt-get update
RUN sudo apt-get install -y git vim wget locales
RUN sudo locale-gen en_US.UTF-8
ENV LANG en_US.UTF-8
ENV LANGUAGE en_US:en
ENV LC_ALL en_US.UTF-8
RUN DEBIAN_FRONTEND=noninteractive TZ=Etc/UTC sudo -E apt-get install -y tzdata

RUN git clone https://github.com/amd/xdna-driver.git --recursive
RUN cd xdna-driver/ && git checkout f60608b
RUN cd xdna-driver/ && git submodule update --recursive
RUN sudo chmod +x /home/riallto/xdna-driver/xrt/src/runtime_src/tools/scripts/xrtdeps.sh
RUN sudo /home/riallto/xdna-driver/xrt/src/runtime_src/tools/scripts/xrtdeps.sh -docker

COPY $BUILD_TEMPDIR/root/debs npu_ubuntu_debs

RUN sudo apt-get install -y dkms udev python3
RUN sudo apt-get install -y python3-pip ocl-icd-opencl-dev uuid-dev libopencv-dev python3-opencv

RUN sudo -E dpkg -i npu_ubuntu_debs/xrt_*-amd64-xrt.deb
RUN sudo -E dpkg -i npu_ubuntu_debs/xrt_plugin*-amdxdna.deb

RUN git clone https://github.com/AMDResearch/Riallto --depth=1

SHELL ["/bin/bash", "-l", "-c"]

# Setup the bashrc environment
RUN echo "source /opt/xilinx/xrt/setup.sh" >> ~/.bashrc
RUN sed -i '6,9d' ~/.bashrc # remove interactive only mode in bashrc
RUN echo "export LD_LIBRARY_PATH=\$LD_LIBRARY_PATH:/opt/xilinx/xrt/lib"

# Setup Riallto
RUN sudo apt-get install -y python3-venv
RUN python3 -m venv ~/riallto_venv
RUN echo "source ~/riallto_venv/bin/activate" >> ~/.bashrc
RUN source ~/.bashrc && cd ~/Riallto/scripts/wsl/ && chmod +x ubuntu_deps.sh && sudo ./ubuntu_deps.sh
RUN source ~/.bashrc && cd ~/Riallto/scripts/wsl/ && python3 -m pip install -r requirements.txt
RUN python3 -m pip install ~/Riallto

# Install MLIR-AIE and Xilinx tools
COPY $BUILD_TEMPDIR/*.tar.gz /home/riallto/
RUN cd ~/Riallto/scripts/wsl/ && chmod +x setup_env.sh && sudo -E ./setup_env.sh /home/riallto/xilinx_tools.tar.gz /home/riallto/pynqMLIR-AIE.tar.gz && sudo chmod -R a+rwx /opt/

# Patch the MLIR-AIE xclbin metadata generation
RUN sudo sed -i '340 i\        "subtype":"DPU",' /opt/mlir-aie/python/aie/compiler/aiecc/main.py

# Setup license
COPY $BUILD_TEMPDIR/Xilinx.lic /opt/
RUN sudo apt-get install -y iproute2

# For external license
RUN sudo -E echo 'export XILINXD_LICENSE_FILE=/opt/Xilinx.lic' >> /opt/mlir_settings.sh
RUN sudo -E echo 'sudo ip link add vmnic0 type dummy || true' >> /opt/mlir_settings.sh
RUN sudo -E echo 'sudo ip link set vmnic0 addr ${LIC_MAC} || true' >> /opt/mlir_settings.sh

# Install Jupyterhub/notebooks
RUN source ~/.bashrc && python3 -m pip install jupyterhub
RUN source ~/.bashrc && python3 -m pip install notebook

#Cleanup a bit
RUN rm -rf /home/riallto/*.tar.gz
ENV SHELL=/bin/bash

CMD ["/bin/bash", "-l", "-c", "jupyterhub-singleuser", "--allow-root"]