Fix build workflow config for pytorch-torchrun-example (kubeflow#2020)

* publish torchrun example via Dockerfile Signed-off-by: PeterWrighten <peterwrighten@gmail.com> * fix context error Signed-off-by: PeterWrighten <peterwrighten@gmail.com> * Update publish-example-images.yaml Co-authored-by: Yuki Iwai <yuki.iwai.tz@gmail.com> Signed-off-by: PeterWrighten <peterwrighten@gmail.com> --------- Signed-off-by: PeterWrighten <peterwrighten@gmail.com> Co-authored-by: Yuki Iwai <yuki.iwai.tz@gmail.com>
johnugeorge · Apr 28, 2024 · 9ed3f19 · 9ed3f19
1 parent 0bb45fe
commit 9ed3f19
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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