update

Author: shink

.github/actions/install-torch/action.yml

@@ -26,5 +26,5 @@ runs:
       id: list-torch-version
       shell: bash
       run: |
-        torch-version=$(python -c "import torch; print(torch.__version__)")
-        echo "torch-version=${torch-version}" >> $GITHUB_OUTPUT
+        torch_version=$(python -c "import torch; print(torch.__version__)")
+        echo "torch-version=${torch_version}" >> $GITHUB_OUTPUT

.github/workflows/_ascend_npu_benchmark.yml

@@ -108,42 +108,20 @@ jobs:
         run: |
           pip install ${{ inputs.torch-artifact }}
 
-      - name: List torch version
-        id: install-torch
-        shell: bash
-        run: |
-          torch_version=$(python -c "import torch; print(torch.__version__)")
-          echo "torch-version=${torch_version}" >> $GITHUB_OUTPUT
-
-      # We must use numpy 1.x
-      - name: Generate pip's constraint file
-        working-directory: ascend_npu
-        run: |
-          cat > constraint.txt << EOF
-          torch==${{ steps.install-torch.outputs.torch-version }}
-          numpy==1.*
-          EOF
-
-      - name: Download ${{ inputs.artifact_name }}
+      - name: Download torch_npu artifact
         uses: actions/download-artifact@v4
         with:
-          name: ${{ inputs.artifact_name }}
+          name: ${{ inputs.torch-npu-artifact }}
           path: ascend_npu
 
       - name: Install torch_npu
         working-directory: ascend_npu
         run: |
-          pip install ${{ inputs.artifact_name }}
-
-      #- name: Install torch_npu dependencies
-      #  working-directory: ascend_npu
-      #  run: |
-      #    curl -fsSL -O https://raw.githubusercontent.com/Ascend/pytorch/refs/heads/master/requirements.txt
-      #    pip install -r requirements.txt
+          pip install ${{ inputs.torch-npu-artifact }}
 
       - name: Install benchmark dependencies
         run: |
-          pip install -r benchmark/requirements.txt --constraint ascend_npu/constraint.txt
+          pip install -r benchmark/requirements.txt numpy==1.*
           python benchmark/install.py --userbenchmark test_bench --continue_on_fail
 
       - name: Install project dependencies

.github/workflows/_ascend_npu_build_torch_npu.yml

@@ -73,6 +73,12 @@ jobs:
           submodules: recursive
           path: torch_npu
 
+      - name: Install pip dependencies
+        working-directory: torch_npu
+        run: |
+          pip install wheel
+          pip install -r requirements.txt
+
       # TODO
       #- name: Install torch
       #  id: install-torch
@@ -90,26 +96,6 @@ jobs:
         run: |
           pip install ${{ inputs.torch-artifact }}
 
-      - name: List torch version
-        id: install-torch
-        shell: bash
-        run: |
-          torch_version=$(python -c "import torch; print(torch.__version__)")
-          echo "torch-version=${torch_version}" >> $GITHUB_OUTPUT
-
-      - name: Generate pip's constraint file
-        working-directory: torch_npu
-        run: |
-          cat > torch-constraint.txt << EOF
-          torch==${{ steps.install-torch.outputs.torch-version }}
-          EOF
-
-      - name: Install pip dependencies
-        working-directory: torch_npu
-        run: |
-          pip install wheel
-          pip install -r requirements.txt --constraint torch-constraint.txt
-
       - name: List Python version
         id: list-py-version
         working-directory: torch_npu

.github/workflows/_ascend_npu_ut.yml

@@ -87,6 +87,13 @@ jobs:
           ref: feat/autoload
           path: torch_npu
 
+      - name: Install pip dependencies
+        working-directory: torch_npu
+        run: |
+          pip install wheel unittest-xml-reporting
+          pip install -r requirements.txt
+          pip install -r test/requirements.txt --no-deps
+
       # TODO
       #- name: Install torch
       #  id: install-torch
@@ -104,33 +111,12 @@ jobs:
         run: |
           pip install ${{ inputs.torch-artifact }}
 
-      - name: List torch version
-        id: install-torch
-        shell: bash
-        run: |
-          torch_version=$(python -c "import torch; print(torch.__version__)")
-          echo "torch-version=${torch_version}" >> $GITHUB_OUTPUT
-
-      - name: Generate pip's constraint file
-        working-directory: torch_npu
-        run: |
-          cat > torch-constraint.txt << EOF
-          torch==${{ steps.install-torch.outputs.torch-version }}
-          EOF
-
       - name: Download torch_npu artifact
         uses: actions/download-artifact@v4
         with:
           name: ${{ inputs.torch-npu-artifact }}
           path: torch_npu
 
-      - name: Install pip dependencies
-        working-directory: torch_npu
-        run: |
-          pip install wheel unittest-xml-reporting
-          pip install -r requirements.txt --constraint torch-constraint.txt
-          pip install -r test/requirements.txt --no-deps
-
       - name: Install torch_npu
         working-directory: torch_npu
         run: |