diff --git a/Dockerfile b/Dockerfile index 07327442fe..973dcedcb5 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,7 +1,7 @@ FROM ubuntu:24.04 ARG DEBIAN_FRONTEND=noninteractive -ARG ROCMVERSION=7.0.1 +ARG ROCMVERSION=7.1.1 ARG compiler_version="" ARG compiler_commit="" ARG CK_SCCACHE="" @@ -13,8 +13,8 @@ ENV DEBIAN_FRONTEND=noninteractive RUN set -xe && \ apt-get update && apt-get install -y --allow-unauthenticated apt-utils wget gnupg2 curl -RUN wget https://repo.radeon.com/amdgpu-install/7.0.1/ubuntu/noble/amdgpu-install_7.0.1.70001-1_all.deb && \ - apt install ./amdgpu-install_7.0.1.70001-1_all.deb -y && \ +RUN wget https://repo.radeon.com/amdgpu-install/7.1.1/ubuntu/noble/amdgpu-install_7.1.1.70101-1_all.deb && \ + apt install ./amdgpu-install_7.1.1.70101-1_all.deb -y && \ apt update && \ apt install python3-setuptools python3-wheel -y && \ apt install rocm-dev -y diff --git a/Dockerfile.compiler b/Dockerfile.compiler index 47bd8294b6..0e2219b7ff 100644 --- a/Dockerfile.compiler +++ b/Dockerfile.compiler @@ -1,4 +1,4 @@ -ARG BASE_DOCKER="rocm/composable_kernel:ck_ub24.04_rocm7.0.1" +ARG BASE_DOCKER="rocm/composable_kernel:ck_ub24.04_rocm7.1.1" FROM $BASE_DOCKER ARG compiler_version="" ARG compiler_commit="" diff --git a/Jenkinsfile b/Jenkinsfile index 45fd576ab6..b8c570b936 100644 --- a/Jenkinsfile +++ b/Jenkinsfile @@ -288,7 +288,7 @@ def getBaseDockerImageName(){ } else{ def ROCM_numeric = parseVersion("${params.ROCMVERSION}") - if ( ROCM_numeric.major <= 7 && ROCM_numeric.minor < 1 ){ + if ( ROCM_numeric.major <= 7 && ROCM_numeric.minor < 2 ){ img = "${env.CK_DOCKERHUB}:ck_ub24.04_rocm${params.ROCMVERSION}" } else{ @@ -434,7 +434,7 @@ def buildDocker(install_prefix){ } catch(Exception ex){ echo "Unable to locate image: ${image_name}. Building image now" - retimage = docker.build("${image_name}", dockerArgs + ' .') + retimage = docker.build("${image_name}", dockerArgs) withDockerRegistry([ credentialsId: "ck_docker_cred", url: "" ]) { retimage.push() } @@ -1121,8 +1121,8 @@ pipeline { description: 'If you want to use a custom docker image, please specify it here (default: leave blank).') string( name: 'ROCMVERSION', - defaultValue: '7.0.1', - description: 'Specify which ROCM version to use: 7.0.1 (default).') + defaultValue: '7.1.1', + description: 'Specify which ROCM version to use: 7.1.1 (default).') string( name: 'COMPILER_VERSION', defaultValue: '', diff --git a/python/ck4inductor/__init__.py b/python/ck4inductor/__init__.py index 0eee25ecaa..089a2d439b 100644 --- a/python/ck4inductor/__init__.py +++ b/python/ck4inductor/__init__.py @@ -6,7 +6,7 @@ def __version__(): import subprocess # needs to be manually updated - rocm_version = "7.0.1" + rocm_version = "7.1.1" hash_width = 6 try: hash = subprocess.check_output("git rev-parse HEAD", shell=True, text=True)[