mount home dir; install Metax Binary Backend

Author: BrianPei

.github/configs/metax.yml

@@ -16,7 +16,7 @@ runner_labels:
 
 container_volumes:
   - /nfs/metax_fs:/nfs/metax_fs
-  # - /home/muxiuser:/home/muxiuser
+  - /home/muxiuser:/home/muxiuser
   # - /home/muxiuser/jinglong/TransformerEngine-FL:/workspace/TransformerEngine-FL
   # - /opt/maca:/opt/maca
   # - /dev/dri:/dev/dri

.github/workflows/unit_tests_common.yml

@@ -182,22 +182,22 @@ jobs:
 
             # echo "===== Step 5: Install Metax Binary Backend ====="
             # # Install prebuilt Metax backend (required for MACA operators)
-            # WHL_PATH="/home/muxiuser/transformer_engine_metax-2.9.0-cp312-cp312-linux_x86_64.whl"
-            # if [ ! -f "$WHL_PATH" ]; then
-            #   echo "ERROR: Wheel file not found at $WHL_PATH"
-            #   echo "Please verify volume mount: -v /home/muxiuser:/home/muxiuser"
-            #   exit 1
-            # fi
+            WHL_PATH="/home/muxiuser/transformer_engine_metax-2.9.0-cp312-cp312-linux_x86_64.whl"
+            if [ ! -f "$WHL_PATH" ]; then
+              echo "ERROR: Wheel file not found at $WHL_PATH"
+              echo "Please verify volume mount: -v /home/muxiuser:/home/muxiuser"
+              exit 1
+            fi
 
             # # Use --no-deps to avoid overwriting Metax-optimized PyTorch
-            # python3 -m pip install "$WHL_PATH" --no-deps --force-reinstall
+            python3 -m pip install "$WHL_PATH" --no-deps --force-reinstall
 
             # echo "===== Step 6: Verify Metax Backend ====="
             # # Ensure transformer_engine_torch is correctly loaded
-            # python3 - <<'EOF'
-            # import transformer_engine_torch as te
-            # print("Backend loaded successfully:", te)
-            # EOF
+            python3 - <<'EOF'
+            import transformer_engine_torch as te
+            print("Backend loaded successfully:", te)
+            EOF
 
             echo "===== Step 7: Install TE-FL Plugin Layer ====="
             # Install TransformerEngine-FL Python layer (plugin logic)