Skip to content

Commit ed61c20

Browse files
Merge pull request #683 from atheo89/RHOAIENG-6445
Update runtime updater workflow to search for the new ROCm runtimes
2 parents 1e851d9 + c813619 commit ed61c20

File tree

1 file changed

+9
-3
lines changed

1 file changed

+9
-3
lines changed

.github/workflows/runtimes-digest-updater-upstream.yaml

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -78,23 +78,29 @@ jobs:
7878
"jupyter/datascience/ubi9-python-3.9/runtime-images/datascience-ubi9-py39.json"
7979
"jupyter/datascience/ubi9-python-3.9/runtime-images/pytorch-ubi9-py39.json"
8080
"jupyter/datascience/ubi9-python-3.9/runtime-images/tensorflow-ubi9-py39.json"
81-
"jupyter/datascience/ubi9-python-3.9/runtime-images/ubi9-py39.json")
81+
"jupyter/datascience/ubi9-python-3.9/runtime-images/ubi9-py39.json"
82+
"jupyter/datascience/ubi9-python-3.9/runtime-images/rocm-tensorflow-ubi9-py39.json"
83+
"jupyter/datascience/ubi9-python-3.9/runtime-images/rocm-pytorch-ubi9-py39.json")
8284
8385
for ((i=0;i<${#PATHS[@]};++i)); do
8486
path=${PATHS[$i]}
8587
img=$(cat ${path} | jq -r '.metadata.image_name')
8688
name=$(echo "$path" | sed 's#.*runtime-images/\(.*\)-py.*#\1#')
8789
py_version=$(echo "$path" | grep -o 'python-[0-9]\.[0-9]')
8890
# Handling specific cases
89-
if [[ $name == *tensorflow* ]]; then
91+
if [[ $name == tensorflow* ]]; then
9092
name="cuda-$name"
9193
elif [[ $name == ubi* ]]; then
9294
name="minimal-$name"
9395
fi
9496
registry=$(echo $img | cut -d '@' -f1)
9597
regex="^runtime-$name-$py_version-${{ env.RELEASE_VERSION_N}}-\d+-${{ steps.hash-n.outputs.HASH_N }}\$"
96-
echo "CHECKING: " $regex
9798
latest_tag=$(skopeo inspect docker://$img | jq -r --arg regex "$regex" '.RepoTags | map(select(. | test($regex))) | .[0]')
99+
echo "CHECKING: " $latest_tag
100+
if [[ -z "$latest_tag" ]]; then
101+
echo "No matching tag found"
102+
exit 1
103+
fi
98104
digest=$(skopeo inspect docker://$registry:$latest_tag | jq .Digest | tr -d '"')
99105
output=$registry@$digest
100106
echo "NEW: " $output

0 commit comments

Comments
 (0)