We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 15cdc3f commit 792016bCopy full SHA for 792016b
.github/workflows/_ascend_npu_benchmark.yml
@@ -69,6 +69,10 @@ jobs:
69
working-directory: ascend_npu
70
run: |
71
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}"
76
77
- name: Checkout benchmark
78
uses: actions/checkout@v4
0 commit comments