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 28587be commit 969dcb2Copy full SHA for 969dcb2
.github/workflows/_ascend_npu_benchmark.yml
@@ -74,10 +74,8 @@ jobs:
74
working-directory: ascend_npu
75
run: |
76
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}"
+ torch_npu_version=$(python -c "import torch_npu; print(torch_npu.__version__)")
+ echo "Installed torch_npu: ${torch_npu_version}"
81
82
- name: Checkout benchmark
83
uses: actions/checkout@v4
0 commit comments