[CI] Copy pre-installed MACA torch packages and install tilelang-metax

Author: wanglei07

.github/workflows/gpu-smoke.yml

@@ -423,11 +423,6 @@ jobs:
           # PR would otherwise reuse.
           skip-atomic-age-trim: "true"
 
-      - name: Set up Python
-        uses: actions/setup-python@v5
-        with:
-          python-version: "3.11"
-
       - name: Resolve runtime state
         env:
           IS_FORK: ${{ needs.security-policy.outputs.is_fork }}
@@ -632,7 +627,26 @@ jobs:
           python -m venv "${VENV_PATH}"
           # shellcheck source=/dev/null
           source "${VENV_PATH}/bin/activate"
+
+          # Copy pre-installed MACA torch packages from conda to venv
+          mkdir -p "${VENV_PATH}/lib/python3.12/site-packages/"
+          cp -rf /opt/conda/lib/python3.12/site-packages/* "${VENV_PATH}/lib/python3.12/site-packages/"
+
           python -m pip install --upgrade pip setuptools wheel --no-user
+
+          # Install test dependencies
+          pip install --no-cache-dir \
+            einops numpy "tqdm>=4.62.3" "typing_extensions>=4.10.0" \
+            cloudpickle ml_dtypes psutil Cython \
+            setuptools wheel ninja
+          pip install --no-cache-dir \
+            "pytest==9.0.2" \
+            "pytest-xdist>=3.0" \
+            "pyyaml>=6.0"
+
+          # Install TileLang (MACA version)
+          pip install --no-cache-dir git+https://github.com/tile-ai/tilelang-metax.git@1efc3ba8a7c48bc97047e562223538d2dcb2222d
+
           if [[ "$IS_FORK" == "true" ]]; then
             FIND_LINKS_FLAG=""
             if [[ -d "${WHEEL_DIR}" ]] && ls "${WHEEL_DIR}"/*.whl >/dev/null 2>&1; then