Skip to content
Merged
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/rocm-ci.yml
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,7 @@ jobs:
ls
- name: Print system info
run: |
rocm-smi
amd-smi list
- uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd # v6.0.2
with:
path: ${{ env.WORKSPACE_DIR }}
Expand Down
6 changes: 3 additions & 3 deletions build/rocm/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -140,8 +140,8 @@ Please follow [ROCm installation guide](https://rocm.docs.amd.com/en/latest/depl
Once installed, verify ROCm installation using:

```Bash
> rocm-smi
============================================ ROCm System Management Interface ============================================
> amd-smi list
============================================ AMD System Management Interface ============================================
====================================================== Concise Info ======================================================
Device Node IDs Temp Power Partitions SCLK MCLK Fan Perf PwrCap VRAM% GPU%
(DID, GUID) (Junction) (Socket) (Mem, Compute, ID)
Expand All @@ -155,7 +155,7 @@ Device Node IDs Temp Power Partitions SCLK
6 8 0x74a1, 53667 42.0°C 140.0W NPS1, SPX, 0 134Mhz 900Mhz 0% auto 750.0W 0% 0%
7 9 0x74a1, 63738 38.0°C 135.0W NPS1, SPX, 0 133Mhz 900Mhz 0% auto 750.0W 0% 0%
==========================================================================================================================
================================================== End of ROCm SMI Log ===================================================
================================================== End of amd-smi list ===================================================
```

### Step 2: Install the Latest Version of JAX
Expand Down
2 changes: 1 addition & 1 deletion ci/utilities/prepare_rocm_tests.sh
Original file line number Diff line number Diff line change
Expand Up @@ -37,7 +37,7 @@ echo "Installed packages:"

"$JAXCI_PYTHON" -c "import jax; print(jax.default_backend()); print(jax.devices()); print(len(jax.devices()))"

rocm-smi
amd-smi list

source ./ci/utilities/rocm_test_env.sh

Expand Down