Skip to content

Commit 969dcb2

Browse files
committed
update
1 parent 28587be commit 969dcb2

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

.github/workflows/_ascend_npu_benchmark.yml

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -74,10 +74,8 @@ jobs:
7474
working-directory: ascend_npu
7575
run: |
7676
pip install ${{ inputs.artifact_name }}
77-
npu_is_available=$(python -c "import torch; print(torch.npu.is_available())")
78-
npu_count=$(python -c "import torch; print(torch.npu.device_count())")
79-
echo "npu is available: ${npu_is_available}"
80-
echo "npu count: ${npu_count}"
77+
torch_npu_version=$(python -c "import torch_npu; print(torch_npu.__version__)")
78+
echo "Installed torch_npu: ${torch_npu_version}"
8179
8280
- name: Checkout benchmark
8381
uses: actions/checkout@v4

0 commit comments

Comments
 (0)