From eda299b84fba61425d11eebf4654e4ecd0c3ba7a Mon Sep 17 00:00:00 2001 From: Nicolas Patry Date: Fri, 7 Jun 2024 20:18:57 +0200 Subject: [PATCH] . --- .github/workflows/build.yaml | 26 ++++++++++++++------------ 1 file changed, 14 insertions(+), 12 deletions(-) diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml index 971a15bdaac..b856517d861 100644 --- a/.github/workflows/build.yaml +++ b/.github/workflows/build.yaml @@ -42,24 +42,26 @@ jobs: run: | case ${{ inputs.hardware }} in cuda) - dockerfile="Dockerfile" - label_extension="" - docker_devices="" - runs_on="[self-hosted, nvidia-gpu , multi-gpu, 4-a10, ci]" + export dockerfile="Dockerfile" + export label_extension="" + export docker_devices="" + export runs_on="[self-hosted, nvidia-gpu , multi-gpu, 4-a10, ci]" ;; rocm) - dockerfile="Dockerfile_amd" - label_extension="-rocm" - docker_devices="/dev/kfd,/dev/dri" - runs_on="ubuntu-latest" + export dockerfile="Dockerfile_amd" + export label_extension="-rocm" + export docker_devices="/dev/kfd,/dev/dri" + export runs_on="ubuntu-latest" ;; rocm) - dockerfile="Dockerfile_intel" - label_extension="-intel" - docker_devices="" - runs_on="ubuntu-latest" + export dockerfile="Dockerfile_intel" + export label_extension="-intel" + export docker_devices="" + export runs_on="ubuntu-latest" ;; esac + echo $dockerfile + echo $label_extension echo "DOCKERFILE=${dockerfile}" >> $GITHUB_ENV echo "LABEL=${label_extension}" >> $GITHUB_ENV echo "DOCKER_DEVICES=${docker_devices}" >> $GITHUB_ENV