mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-12 09:16:52 +00:00
7bcaa73a3a
[CK] Fix for hipblaslt error in PyTorch Dockerfile ## Motivation This PR fixes the hipblaslt client build failures that occur when building the PyTorch Docker image, which are currently causing failures in CI. ## Technical Details - Correctly reset the working directory to tmp/ - Added --use-system-packages to the install.sh to use system installed laplack packages, as hard-coded paths were not being built. ## Test Plan Locally built the Docker image using the Dockerfile. ## Test Result Image was successfully built. ## Submission Checklist - [x] Look over the contributing guidelines at https://github.com/ROCm/ROCm/blob/develop/CONTRIBUTING.md#pull-requests. Co-authored-by: Illia Silin <98187287+illsilin@users.noreply.github.com>
34 lines
1.5 KiB
Docker
34 lines
1.5 KiB
Docker
ARG BASE_DOCKER="rocm/pytorch-nightly:latest"
|
|
FROM $BASE_DOCKER
|
|
ARG CK_PYTORCH_BRANCH="develop"
|
|
RUN groupadd -g 109 render && \
|
|
usermod -u 1001 jenkins && \
|
|
groupmod -g 1001 jenkins && \
|
|
cd /tmp/pytorch && \
|
|
rm -rf build && \
|
|
cd /tmp/pytorch/third_party && \
|
|
rm -rf composable_kernel && \
|
|
git clone -b "$CK_PYTORCH_BRANCH" https://github.com/ROCm/composable_kernel.git && \
|
|
cd /tmp/pytorch/third_party/aiter/3rdparty && \
|
|
rm -rf composable_kernel && \
|
|
git clone -b "$CK_PYTORCH_BRANCH" https://github.com/ROCm/composable_kernel.git && \
|
|
cd /tmp/pytorch/third_party/fbgemm/external && \
|
|
rm -rf composable_kernel && \
|
|
git clone -b "$CK_PYTORCH_BRANCH" https://github.com/ROCm/composable_kernel.git && \
|
|
cd /tmp/pytorch/third_party/flash-attention/csrc && \
|
|
rm -rf composable_kernel && \
|
|
git clone -b "$CK_PYTORCH_BRANCH" https://github.com/ROCm/composable_kernel.git && \
|
|
chown -R jenkins:jenkins /tmp/pytorch && \
|
|
chmod -R a+rwx /tmp/pytorch && \
|
|
sudo usermod -aG irc jenkins && \
|
|
#install hipblaslt
|
|
cd /tmp && \
|
|
git clone --no-checkout --filter=blob:none https://github.com/ROCm/rocm-libraries.git && \
|
|
cd rocm-libraries && \
|
|
git checkout develop && \
|
|
git sparse-checkout init --cone && \
|
|
git sparse-checkout set projects/hipblaslt shared/origami && \
|
|
cd projects/hipblaslt && \
|
|
git show --oneline -s && \
|
|
CPLUS_INCLUDE_PATH="/opt/amdgpu/include/" ./install.sh -idc --use-system-packages --architecture="gfx942;gfx950" -j 128 --skip_rocroller
|