From 14eeaeb0408457a9b990195b3034eb4ea34ebd90 Mon Sep 17 00:00:00 2001 From: PeterWright Date: Sat, 9 Mar 2024 21:06:40 +0900 Subject: [PATCH] Fix build workflow config for pytorch-torchrun-example (#2020) * publish torchrun example via Dockerfile Signed-off-by: PeterWrighten * fix context error Signed-off-by: PeterWrighten * Update publish-example-images.yaml Co-authored-by: Yuki Iwai Signed-off-by: PeterWrighten --------- Signed-off-by: PeterWrighten Co-authored-by: Yuki Iwai --- .github/workflows/publish-example-images.yaml | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/.github/workflows/publish-example-images.yaml b/.github/workflows/publish-example-images.yaml index dfe2ae9004..7bbf3e2170 100644 --- a/.github/workflows/publish-example-images.yaml +++ b/.github/workflows/publish-example-images.yaml @@ -38,8 +38,9 @@ jobs: dockerfile: examples/pytorch/elastic/imagenet/Dockerfile - component-name: pytorch-elastic-example-echo dockerfile: examples/pytorch/elastic/echo/Dockerfile - - component-name: pytorch-cpu-demo-torchrun-cpu + - component-name: pytorch-torchrun-cpu dockerfile: examples/pytorch/cpu-demo/Dockerfile + context: examples/pytorch/cpu-demo - component-name: xgboost-dist-iris dockerfile: examples/xgboost/xgboost-dist/Dockerfile context: examples/xgboost/xgboost-dist