Skip to content

Commit 792016b

Browse files
committed
update
1 parent 15cdc3f commit 792016b

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

.github/workflows/_ascend_npu_benchmark.yml

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -69,6 +69,10 @@ jobs:
6969
working-directory: ascend_npu
7070
run: |
7171
pip install ${{ inputs.artifact_name }}
72+
npu_is_available=$(python -c "import torch; print(torch.npu.is_available()")
73+
npu_count=$(python -c "import torch; print(torch.npu.device_count()")
74+
echo "npu is available: ${npu_is_available}"
75+
echo "npu count: ${npu_count}"
7276
7377
- name: Checkout benchmark
7478
uses: actions/checkout@v4

0 commit comments

Comments
 (0)