update yml

dptech-corp · Jul 12, 2024 · f26440a · f26440a
1 parent 17848b4
commit f26440a
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/.github/workflows/docker_rdma_latest.yml b/.github/workflows/docker_rdma_latest.yml
@@ -5,6 +5,7 @@ on:
     branches:
     - main
     - cuda12
+    - cuda12-fa3
 
 jobs:
   docker:
@@ -60,5 +61,5 @@ jobs:
           context: ./docker/rdma/
           push: true
           tags: |
-            dptechnology/unicore:latest-pytorch2.3.0-cuda12.4-rdma
-            dp-ve-registry-cn-beijing.cr.volces.com/dplc/unicore:latest-pytorch2.3.0-cuda12.4-rdma
+            dptechnology/unicore:latest-pytorch2.3.0-cuda12.4-rdma-fa3
+            dp-ve-registry-cn-beijing.cr.volces.com/dplc/unicore:latest-pytorch2.3.0-cuda12.4-rdma-fa3