Skip to content

Commit ec49192

Browse files
committed
NO-JIRA: fix(ci): update the way we check for the rocm version
With the introduction of the 2025.1 images, we changed the way we install the ROCm on our images. Instead of installing the main `rocm` metapackage, we install directly just it's subpackages ecluding the installation of the `migraphx` and `mivisionx` packages to reduce the size of the result ROCm images. This was a deliberate decision and so we need to amend the way how to check the ROCm version on our images. For the record: * 0d262e2#diff-8ddc31b6fcd6307008d6abf6aef98768f83a7a94cb09a2eee2b3e1ff04788f20R40-R42
1 parent 96dab15 commit ec49192

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ci/check-software-versions.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -148,7 +148,7 @@ def process_dependency_item(item, container_id, annotation_type):
148148

149149
command_mapping = {
150150
"PyTorch": ["/bin/bash", "-c", f"pip show torch | grep 'Version: '"],
151-
"ROCm": ["/bin/bash", "-c", "rpm -q --queryformat '%{VERSION}\n' rocm"],
151+
"ROCm": ["/bin/bash", "-c", "rpm -q --queryformat '%{VERSION}\n' rocm-core"],
152152
"ROCm-PyTorch": ["/bin/bash", "-c", "pip show torch | grep 'Version: ' | grep rocm"],
153153
"ROCm-TensorFlow": ["/bin/bash", "-c", "pip show tensorflow-rocm | grep 'Version: '"],
154154
"TensorFlow": ["/bin/bash", "-c", "pip show tensorflow | grep 'Version: '"],

0 commit comments

Comments
 (0)