Fix community uploads (#10389)

Oneflow-Inc · Dec 28, 2023 · a0f2122 · a0f2122
1 parent 92788cb
commit a0f2122
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/.github/workflows/community_release.yml b/.github/workflows/community_release.yml
@@ -28,6 +28,7 @@ jobs:
     with:
       is_priv: true
       branch: ${{ inputs.priv_branch || 'main' }}
+      upload_override_branch: "community"
       cuda_cmake_cache: cmake/caches/ci/release/cuda_community.cmake
     secrets:
       ONEFLOW_PRIV_ORG: ${{ secrets.ONEFLOW_PRIV_ORG }}

diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml
@@ -23,6 +23,9 @@ on:
         required: false
         type: string
         default: "main"
+      upload_override_branch:
+        required: false
+        type: string
       cuda_cmake_cache:
         required: false
         type: string
@@ -135,8 +138,8 @@ jobs:
         if: ${{ inputs.is_priv }}
         run: |
           GITHUB_SHA=$(git rev-parse HEAD)
-          echo "OSS_DIR=branch/${{ inputs.branch }}/${{ matrix.entry }}/${GITHUB_SHA}" >> $GITHUB_ENV
-          echo "GITHUB_REF_NAME=${{ inputs.branch }}" >> $GITHUB_ENV
+          echo "OSS_DIR=branch/${{ inputs.upload_override_branch || inputs.branch }}/${{ matrix.entry }}/${GITHUB_SHA}" >> $GITHUB_ENV
+          echo "GITHUB_REF_NAME=${{ inputs.upload_override_branch || inputs.branch }}" >> $GITHUB_ENV
           echo "GITHUB_SHA=${GITHUB_SHA}" >> $GITHUB_ENV
           echo "ONEFLOW_OSS_BUCKET=${{ secrets.ONEFLOW_PRIV_OSS_BUCKET }}" >> $GITHUB_ENV
       - name: Print env