Dockerfile

# vim: nospell

# https://hub.docker.com/r/nvidia/cuda/
FROM nvidia/cuda:10.2-devel-ubuntu18.04
LABEL maintainer "Trevor L. McDonell <trevor.mcdonell@gmail.com>"

ARG DEBIAN_FRONTEND=noninteractive
ARG PREFIX=/opt/accelerate-llvm

ENV LANG C.UTF-8
ENV LC_ALL C.UTF-8
ENV PATH /root/.cabal/bin:/root/.local/bin:${PATH}
ENV LD_LIBRARY_PATH /usr/local/cuda/lib64:/usr/local/cuda/nvvm/lib64:${LD_LIBRARY_PATH}
RUN ln -s /usr/local/cuda/lib64/stubs/libcuda.so /usr/local/cuda/lib64/libcuda.so.1

RUN apt-get update \
 && apt-get install -y software-properties-common

RUN add-apt-repository -y ppa:hvr/ghc \
 && apt-get update \
 && apt-get install -y curl netbase pkg-config wget

RUN curl -sSL https://get.haskellstack.org/ | sh

# Buggy versions of ld.bfd fail to link some Haskell packages:
# https://sourceware.org/bugzilla/show_bug.cgi?id=17689.
RUN update-alternatives --install "/usr/bin/ld" "ld" "/usr/bin/ld.gold" 20
RUN update-alternatives --install "/usr/bin/ld" "ld" "/usr/bin/ld.bfd" 10

# Install LLVM
RUN wget -O - http://apt.llvm.org/llvm-snapshot.gpg.key | apt-key add - \
 && add-apt-repository "deb http://apt.llvm.org/bionic/ llvm-toolchain-bionic main" \
 && apt-get update \
 && apt-get install -y llvm-9-dev

# GHC requires a specific LLVM version on the system PATH for its LLVM backend.
# The version is tracked here:
# https://ghc.haskell.org/trac/ghc/wiki/Commentary/Compiler/Backends/LLVM/Installing
# RUN update-alternatives --install "/usr/bin/llc" "llc" "/usr/bin/llc-6" 50
# RUN update-alternatives --install "/usr/bin/opt" "opt" "/usr/bin/opt-6" 50

# Setup stack and build dependencies
WORKDIR ${PREFIX}
COPY ./README.md                                            ${PREFIX}/
COPY ./LICENSE                                              ${PREFIX}/
COPY ./stack-8.10.yaml                                      ${PREFIX}/stack.yaml
COPY ./accelerate-llvm/accelerate-llvm.cabal                ${PREFIX}/accelerate-llvm/
COPY ./accelerate-llvm-native/accelerate-llvm-native.cabal  ${PREFIX}/accelerate-llvm-native/
COPY ./accelerate-llvm-ptx/accelerate-llvm-ptx.cabal        ${PREFIX}/accelerate-llvm-ptx/

# GHC 8.4 requires libffi7
ENV LD_LIBRARY_PATH $(stack exec ghc -- --print-lib)/rts:${LD_LIBRARY_PATH}

# Setup build environment
RUN stack --no-terminal --color never setup
RUN stack --no-terminal --color never build --only-snapshot

# Copy over the actual source files and build
COPY ./accelerate-llvm        ${PREFIX}/accelerate-llvm
COPY ./accelerate-llvm-native ${PREFIX}/accelerate-llvm-native
COPY ./accelerate-llvm-ptx    ${PREFIX}/accelerate-llvm-ptx

RUN stack --no-terminal --color never build accelerate-llvm
RUN stack --no-terminal --color never build accelerate-llvm-native
RUN stack --no-terminal --color never build accelerate-llvm-ptx

# libcuda.so.1 is part of the nvidia driver. We need this hack to complete the
# build because dockerhub is not running with the nvidia-docker tool, which I
# believe is what is meant to provide this lib in a way allowing passthrough to
# the actual driver library running on host machine.
#
# https://github.com/tmcdonell/cuda/issues/55
RUN rm /usr/local/cuda/lib64/libcuda.so.1

CMD ["bash"]