ci: Fix release

Signed-off-by: oliver könig <okoenig@nvidia.com>

Author: ko3n1g

.gitlab/stages/03.publish.yml

@@ -4,7 +4,7 @@
     - if: $CI_COMMIT_BRANCH == $CI_DEFAULT_BRANCH && $PUBLISH == "yes" && $PUBLISH_SCOPE == "code-freeze"
       when: manual
     - when: never
-  
+
 .publish_common_release:
   stage: publish
   rules:
@@ -13,7 +13,7 @@
     - if: $PUBLISH == "yes" && $PUBLISH_SCOPE == "release"
       when: manual
       variables:
-        PUBLISH_DRYRUN: "yes"
+        PUBLISH_DRYRUN: 'yes'
     - when: never
 
 publish:release_branch:
@@ -22,13 +22,13 @@ publish:release_branch:
   needs: [test:build_image]
   tags: [mcore-docker-node-small]
   variables:
-    GIT_STRATEGY: "none"
+    GIT_STRATEGY: 'none'
   script:
     - git fetch origin $CI_DEFAULT_BRANCH
     - git config --global user.email "mcore-bot@nvidia.com"
     - git config --global user.name "Mcore Bot"
     - git remote set-url origin "https://gitlab-ci-token:${PAT}@${GITLAB_ENDPOINT}/$CI_PROJECT_NAMESPACE/megatron-lm.git"
-    - sed -i "/^PRE_RELEASE/c\PRE_RELEASE = ''" megatron/core/package_info.py 
+    - sed -i "/^PRE_RELEASE/c\PRE_RELEASE = ''" megatron/core/package_info.py
     - VERSION=$(python -c "from megatron import core; print(core.__version__)")
     - RELEASE_BRANCH=core_r$VERSION
     - git switch --force-create $RELEASE_BRANCH origin/$CI_DEFAULT_BRANCH
@@ -46,11 +46,15 @@ publish:release_branch:
       }'
     - >
       curl -X POST -H "Content-type: application/json" --data "$MESSAGE" ${MCORE_NOTIFICATION_HOOK_MAIN}
-    - git switch --force-create bot/chore/bump-version 
-    - git add megatron/core/package_info.py 
+
+
+    - git switch --force-create bot/chore/bump-version
+    - git add megatron/core/package_info.py
     - >
       git commit -m "chore: adjust version version"
-    - git push -u origin bot/chore/bump-version 
+
+
+    - git push -u origin bot/chore/bump-version
     - >
       curl \
         --header "PRIVATE-TOKEN: $PAT" \
@@ -64,28 +68,47 @@ publish:pypi_build_wheel:
   extends: [test:pypi_build_wheel, .publish_common_release]
   dependencies: []
   variables:
-    PUBLISH_DRYRUN: "no"
-  
+    PUBLISH_DRYRUN: 'no'
+
 publish:pypi_test_wheel:
   extends: [test:pypi_test_wheel, .publish_common_release]
   needs: [publish:pypi_build_wheel]
   variables:
-    PUBLISH_DRYRUN: "no"
-  
+    PUBLISH_DRYRUN: 'no'
+
 publish:pypi_push_wheel:
   extends: [test:pypi_push_wheel, .publish_common_release]
   needs: [publish:pypi_test_wheel]
+  dependencies: [publish:pypi_test_wheel]
   variables:
-    PUBLISH_DRYRUN: "no"
+    PUBLISH_DRYRUN: 'no'
 
 publish:gh_release:
   extends: [test:gh_release, .publish_common_release]
-  dependencies: []
+  dependencies: [publish:pypi_test_wheel]
+  needs: [publish:pypi_test_wheel]
   variables:
-    PUBLISH_DRYRUN: "no"
-      
+    PUBLISH_DRYRUN: 'no'
+
 publish:notify_release:
   needs: [publish:pypi_push_wheel, publish:gh_release]
   extends: [test:notify_release, .publish_common_release]
   variables:
-    PUBLISH_DRYRUN: "no"
+    PUBLISH_DRYRUN: 'no'
+
+publish:docs:
+  extends: [.publish_common_release]
+  image: ${UTILITY_IMAGE}:${CI_PIPELINE_ID}
+  tags: [mcore-docker-node-small]
+  script:
+    - cd ..
+    - rm -rf documentation && git clone https://gitlab-ci-token:${PROJECT_ACCESS_TOKEN_MCORE}@${GITLAB_ENDPOINT}/nemo-megatron-core-tme/documentation.git
+    - cd documentation/megatron-lm
+    - git fetch
+    - git checkout $CI_COMMIT_SHA
+    - cd ..
+    - git add megatron-lm
+    - >
+      git commit -m 'feat: Bump mcore'
+
+    - git push