diff --git a/.github/workflows/_ascend_npu_benchmark.yml b/.github/workflows/_ascend_npu_benchmark.yml index 74e4fef..fbbfa65 100644 --- a/.github/workflows/_ascend_npu_benchmark.yml +++ b/.github/workflows/_ascend_npu_benchmark.yml @@ -73,7 +73,7 @@ jobs: - name: Checkout benchmark uses: actions/checkout@v4 with: - repository: pytorch/benchmark + repository: cllouud/benchmark path: benchmark # TODO diff --git a/.github/workflows/_ascend_npu_build_torch.yml b/.github/workflows/_ascend_npu_build_torch.yml index 05113a1..2597e68 100644 --- a/.github/workflows/_ascend_npu_build_torch.yml +++ b/.github/workflows/_ascend_npu_build_torch.yml @@ -82,7 +82,7 @@ jobs: run: | python setup.py build bdist_wheel env: - MAX_JOBS: 8 + MAX_JOBS: 16 - name: List distribution package id: list-dist diff --git a/.github/workflows/_ascend_npu_build_torch_npu.yml b/.github/workflows/_ascend_npu_build_torch_npu.yml index a9cdaec..749456e 100644 --- a/.github/workflows/_ascend_npu_build_torch_npu.yml +++ b/.github/workflows/_ascend_npu_build_torch_npu.yml @@ -1,5 +1,6 @@ name: '_ascend_npu_build' + on: workflow_call: inputs: @@ -104,7 +105,7 @@ jobs: bash ci/build.sh --python=${{ steps.list-py-version.outputs.py-version }} env: _GLIBCXX_USE_CXX11_ABI: 1 # https://gitee.com/ascend/pytorch/issues/IBJFZY - MAX_JOBS: 8 + MAX_JOBS: 16 - name: List distribution package id: list-dist @@ -123,6 +124,7 @@ jobs: - name: List torch_npu version id: list-version run: | + export TORCH_DEVICE_BACKEND_AUTOLOAD=0 torch_version=$(python -c "import torch; print(torch.__version__)") torch_npu_version=$(python -c "import torch_npu; print(torch_npu.__version__)") echo "torch version: ${torch_version}" diff --git a/.github/workflows/_ascend_npu_ut.yml b/.github/workflows/_ascend_npu_ut.yml index 67ec667..1fad8c5 100644 --- a/.github/workflows/_ascend_npu_ut.yml +++ b/.github/workflows/_ascend_npu_ut.yml @@ -65,7 +65,7 @@ jobs: - name: Checkout torch_npu uses: actions/checkout@v4 with: - repository: Ascend/pytorch + repository: cllouud/pytorch path: torch_npu - name: Install pip dependencies