Skip to content
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion .github/workflows/_ascend_npu_benchmark.yml
Original file line number Diff line number Diff line change
Expand Up @@ -73,7 +73,7 @@ jobs:
- name: Checkout benchmark
uses: actions/checkout@v4
with:
repository: pytorch/benchmark
repository: cllouud/benchmark
path: benchmark

# TODO
Expand Down
2 changes: 1 addition & 1 deletion .github/workflows/_ascend_npu_build_torch.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
4 changes: 3 additions & 1 deletion .github/workflows/_ascend_npu_build_torch_npu.yml
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
name: '_ascend_npu_build'


on:
workflow_call:
inputs:
Expand Down Expand Up @@ -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
Expand All @@ -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}"
Expand Down
2 changes: 1 addition & 1 deletion .github/workflows/_ascend_npu_ut.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
Loading