diff --git a/Dockerfile.pytorch b/Dockerfile.pytorch index 1b71b00fbb..4533166c06 100644 --- a/Dockerfile.pytorch +++ b/Dockerfile.pytorch @@ -20,4 +20,13 @@ RUN groupadd -g 109 render && \ 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 + sudo usermod -aG irc jenkins && \ + #install hipblaslt + 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 --logic-yaml-filter gfx950/*/* --architecture="gfx942;gfx950" -j 128 --skip_rocroller diff --git a/Jenkinsfile b/Jenkinsfile index a2e5b3d20b..45fd576ab6 100644 --- a/Jenkinsfile +++ b/Jenkinsfile @@ -447,7 +447,7 @@ def get_docker_options(){ dockerOpts = "--network=host --group-add video --group-add render --cap-add=SYS_PTRACE --security-opt seccomp=unconfined" } else{ //only add kfd and dri paths if you actually going to run somthing on GPUs - dockerOpts = "--network=host --device=/dev/kfd --device=/dev/dri --group-add video --group-add render --group-add irc --cap-add=SYS_PTRACE --security-opt seccomp=unconfined" + dockerOpts = "--network=host --device=/dev/kfd --device=/dev/dri --group-add video --group-add render --cap-add=SYS_PTRACE --security-opt seccomp=unconfined" } if (params.COMPILER_VERSION == "amd-staging" || params.COMPILER_VERSION == "amd-mainline" || params.COMPILER_COMMIT != ""){ // the --env COMPRESSED_BUNDLE_FORMAT_VERSION=2 env variable is required when building code with offload-compress flag with @@ -1003,7 +1003,7 @@ def run_aiter_tests(Map conf=[:]){ checkout scm //use the latest pytorch image def image = "${env.CK_DOCKERHUB_PRIVATE}:ck_aiter" - def dockerOpts=get_docker_options() + def dockerOpts=get_docker_options() + ' --group-add irc ' gitStatusWrapper(credentialsId: "${env.ck_git_creds}", gitHubContext: "${env.STAGE_NAME}", account: 'ROCm', repo: 'composable_kernel') { try @@ -1055,7 +1055,7 @@ def run_pytorch_tests(Map conf=[:]){ checkout scm //use the latest pytorch-nightly image def image = "${env.CK_DOCKERHUB}:ck_pytorch" - def dockerOpts=get_docker_options() + def dockerOpts=get_docker_options() + ' --group-add irc ' gitStatusWrapper(credentialsId: "${env.ck_git_creds}", gitHubContext: "${env.STAGE_NAME}", account: 'ROCm', repo: 'composable_kernel') { try