27 lines
1.6 KiB
Docker
27 lines
1.6 KiB
Docker
ARG BASE_DOCKER="rocm/composable_kernel:ck_ub22.04_rocm6.3"
|
|
FROM $BASE_DOCKER
|
|
ARG compiler_version=""
|
|
ARG compiler_commit=""
|
|
|
|
# Add alternative compilers, if necessary
|
|
ENV compiler_version=$compiler_version
|
|
ENV compiler_commit=$compiler_commit
|
|
RUN sh -c "echo compiler version = '$compiler_version'" && \
|
|
sh -c "echo compiler commit = '$compiler_commit'"
|
|
|
|
RUN if ( [ "$compiler_version" = "amd-staging" ] || [ "$compiler_version" = "amd-mainline" ] ) && [ "$compiler_commit" = "" ]; then \
|
|
git clone -b "$compiler_version" https://github.com/ROCm/llvm-project.git && \
|
|
cd llvm-project && mkdir build && cd build && \
|
|
cmake -DCMAKE_INSTALL_PREFIX=/opt/rocm/llvm -DCMAKE_BUILD_TYPE=Release -DLLVM_ENABLE_ASSERTIONS=1 -DLLVM_TARGETS_TO_BUILD="AMDGPU;X86" -DLLVM_ENABLE_PROJECTS="clang;lld" -DLLVM_ENABLE_RUNTIMES="compiler-rt" ../llvm && \
|
|
make -j 16 ; \
|
|
else echo "using the release compiler"; \
|
|
fi
|
|
|
|
RUN if ( [ "$compiler_version" = "amd-staging" ] || [ "$compiler_version" = "amd-mainline" ] ) && [ "$compiler_commit" != "" ]; then \
|
|
git clone -b "$compiler_version" https://github.com/ROCm/llvm-project.git && \
|
|
cd llvm-project && git checkout "$compiler_commit" && echo "checking out commit $compiler_commit" && mkdir build && cd build && \
|
|
cmake -DCMAKE_INSTALL_PREFIX=/opt/rocm/llvm -DCMAKE_BUILD_TYPE=Release -DLLVM_ENABLE_ASSERTIONS=1 -DLLVM_TARGETS_TO_BUILD="AMDGPU;X86" -DLLVM_ENABLE_PROJECTS="clang;lld" -DLLVM_ENABLE_RUNTIMES="compiler-rt" ../llvm && \
|
|
make -j 16 ; \
|
|
else echo "using the release compiler"; \
|
|
fi
|