Skip to content

Commit dd41698

Browse files
LukaPrebilmertalev
andauthored
fix(ml): Upgrade ONNX Runtime to v1.22.1 to fix ROCm build failures (#24045)
* fix: update ONNX runtime version to 1.21.0 to fix the failing checksum of 1.20.1 * update patch * update to 1.22.1 --------- Co-authored-by: mertalev <101130780+mertalev@users.noreply.github.com>
1 parent 8321c27 commit dd41698

File tree

2 files changed

+10
-10
lines changed

2 files changed

+10
-10
lines changed

machine-learning/Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ FROM builder-cpu AS builder-rknn
2525
FROM rocm/dev-ubuntu-22.04:6.4.3-complete@sha256:1f7e92ca7e3a3785680473329ed1091fc99db3e90fcb3a1688f2933e870ed76b AS builder-rocm
2626

2727
# renovate: datasource=github-releases depName=Microsoft/onnxruntime
28-
ARG ONNXRUNTIME_VERSION="v1.20.1"
28+
ARG ONNXRUNTIME_VERSION="v1.22.1"
2929
WORKDIR /code
3030

3131
RUN apt-get update && apt-get install -y --no-install-recommends wget git python3.10-venv
Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,13 @@
11
diff --git a/cmake/CMakeLists.txt b/cmake/CMakeLists.txt
2-
index d90a2a355..bb1a7de12 100644
2+
index 2714e6f59..a69da76b4 100644
33
--- a/cmake/CMakeLists.txt
44
+++ b/cmake/CMakeLists.txt
5-
@@ -295,7 +295,7 @@ if (onnxruntime_USE_ROCM)
5+
@@ -338,7 +338,7 @@ if (onnxruntime_USE_ROCM)
6+
if (ROCM_VERSION_DEV VERSION_LESS "6.2")
7+
message(FATAL_ERROR "CMAKE_HIP_ARCHITECTURES is not set when ROCm version < 6.2")
8+
else()
9+
- set(CMAKE_HIP_ARCHITECTURES "gfx908;gfx90a;gfx1030;gfx1100;gfx1101;gfx940;gfx941;gfx942;gfx1200;gfx1201")
10+
+ set(CMAKE_HIP_ARCHITECTURES "gfx900;gfx908;gfx90a;gfx1030;gfx1100;gfx1101;gfx1102;gfx940;gfx941;gfx942;gfx1200;gfx1201")
11+
endif()
612
endif()
7-
8-
if (NOT CMAKE_HIP_ARCHITECTURES)
9-
- set(CMAKE_HIP_ARCHITECTURES "gfx908;gfx90a;gfx1030;gfx1100;gfx1101;gfx940;gfx941;gfx942;gfx1200;gfx1201")
10-
+ set(CMAKE_HIP_ARCHITECTURES "gfx900;gfx908;gfx90a;gfx1030;gfx1100;gfx1101;gfx1102;gfx940;gfx941;gfx942;gfx1200;gfx1201")
11-
endif()
12-
13-
file(GLOB rocm_cmake_components ${onnxruntime_ROCM_HOME}/lib/cmake/*)
13+

0 commit comments

Comments
 (0)