From 224e01716b2b48c0b4bbc80d1c2b5610b7ca9554 Mon Sep 17 00:00:00 2001 From: ydshieh Date: Tue, 27 Feb 2024 17:03:11 +0800 Subject: [PATCH] use matrix --- .github/workflows/run_build_and_test.yml | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/.github/workflows/run_build_and_test.yml b/.github/workflows/run_build_and_test.yml index b2e9e4e2..f3edd054 100644 --- a/.github/workflows/run_build_and_test.yml +++ b/.github/workflows/run_build_and_test.yml @@ -7,8 +7,14 @@ on: jobs: build-push-image: + strategy: + fail-fast: false + matrix: + torch: [2.1] + transformers: [4.38.1] + python: [310] uses: ./.github/workflows/build_and_test.yml secrets: inherit with: - dockerfile_path: ./containers/pytorch/training/gpu/2.1/transformers/4.38.1/py310/Dockerfile - docker_image_tag: huggingface-pytorch-training-gpu-2.1.transformers.4.38.1.py310:latest + dockerfile_path: ./containers/pytorch/training/gpu/$ {{ matrix.torch }}/transformers/$ {{ matrix.transformers }}/py$ {{ matrix.python }}/Dockerfile + docker_image_tag: huggingface-pytorch-training-gpu-$ {{ matrix.torch }}.transformers.$ {{ matrix.transformers }}.py$ {{ matrix.python }}:latest