use actions/checkout@v4

Author: zhoujiamei

.github/workflows/unit_tests_common.yml

@@ -92,11 +92,12 @@ jobs:
     steps:
 
 
-      # - name: Checkout source code
-      #   uses: actions/checkout@v4
-      #   with:
-      #     set-safe-directory: true
-      #     path: TransformerEngine-FL
+      - name: Checkout source code
+        uses: actions/checkout@v4
+        with:
+          repository: flagos-ai/TransformerEngine-FL
+          path: TransformerEngine-FL
+          submodules: recursive
 
       - name: Create and Start Clean Container
         run: |
@@ -116,6 +117,8 @@ jobs:
             --device=/dev/mxcd \
             --device=/dev/infiniband \
             -v /nfs/metax_fs:/nfs/metax_fs \
+            -v ${{ github.workspace }}/TransformerEngine-FL:/workspace/TransformerEngine-FL \
+            -w /workspace/TransformerEngine-FL \
             ${{ inputs.image }} \
             /bin/bash
             
@@ -152,11 +155,11 @@ jobs:
             echo '===== 2. git clone ====='
             apt-get update
             apt-get install -y git
-            #http1.1
-            git config --global http.version HTTP/1.1
-            git config --global http.postBuffer 524288000
+            # #http1.1
+            # git config --global http.version HTTP/1.1
+            # git config --global http.postBuffer 524288000
 
-            git clone https://github.com/flagos-ai/TransformerEngine-FL.git
+            # git clone https://github.com/flagos-ai/TransformerEngine-FL.git
             python3 -m pip install nvdlfw-inspect --no-deps || true
             python3 -m pip install --upgrade pip setuptools wheel --user