Skip to content

Commit 1bd07a6

Browse files
Merge pull request #984 from jstourac/updateRocmCheck
NO-JIRA: fix(ci): update the way we check for the rocm version
2 parents 96dab15 + ec49192 commit 1bd07a6

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)