Tweaks to docker scripts [skip ci] (#28)

docker/build-docker-release.sh

@@ -12,18 +12,22 @@ build-release()
     CONTAINER=$1
     ROCM_VERSION=$2
     CODE_VERSION=$3
-    docker run -it --rm -v ${PWD}:/home/omnitrace --env ROCM_VERSION=${ROCM_VERSION} --env VERSION=${CODE_VERSION} ${CONTAINER} /home/omnitrace/scripts/build-release.sh
+    MPI=$4
+    docker run -it --rm -v ${PWD}:/home/omnitrace --env ROCM_VERSION=${ROCM_VERSION} --env VERSION=${CODE_VERSION} --env MPI=${MPI} ${CONTAINER} /home/omnitrace/scripts/build-release.sh
 }
 
 : ${DISTRO:=ubuntu}
 : ${VERSIONS:=20.04 18.04}
+: ${ROCM_VERSIONS:=5.0 4.5 4.3}
+: ${MPI:=0}
 
 CODE_VERSION=$(cat VERSION)
 
 for VERSION in ${VERSIONS}
 do
     TAG=${DISTRO}-${VERSION}
-    build-release jrmadsen/omnitrace-${TAG}-rocm-4.5   4.5.0 ${CODE_VERSION}
-    build-release jrmadsen/omnitrace-${TAG}-rocm-4.3   4.3.0 ${CODE_VERSION}
-    build-release jrmadsen/omnitrace-${TAG}-rocm-4.3.1 4.3.1 ${CODE_VERSION}
+    for ROCM_VERSION in ${ROCM_VERSIONS}
+    do
+        build-release jrmadsen/omnitrace-${TAG}-rocm-${ROCM_VERSION} ${ROCM_VERSION} ${CODE_VERSION} ${MPI}
+    done
 done