add mcore installation to ci

Signed-off-by: jiemingz <jiemingz@nvidia.com>
JimmyZhang12 · Jan 23, 2024 · 9da045a · 9da045a
1 parent 4f687c0
commit 9da045a
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/Jenkinsfile b/Jenkinsfile
@@ -81,14 +81,14 @@ pipeline {
 
     // pip package should be working with main, if not we can update the commit here
     // until the pip package is updated
-    // stage('Megatron Core installation') {
-    //   steps {
-    //      sh 'git clone https://github.com/NVIDIA/Megatron-LM.git && \
-    //          cd Megatron-LM && \
-    //          git checkout 973330e9c3681604703bf1eb6b5a265d1b9b9b38 && \
-    //          pip install .'
-    //   }
-    // }
+    stage('Megatron Core installation') {
+      steps {
+         sh 'git clone https://github.com/NVIDIA/Megatron-LM.git && \
+             cd Megatron-LM && \
+             git checkout 973330e9c3681604703bf1eb6b5a265d1b9b9b38 && \
+             pip install .'
+      }
+    }
 
     stage('PyTorch Lightning version') {
       steps {