diff --git a/.github/workflows/rocm-ci.yml b/.github/workflows/rocm-ci.yml index a66ea231ef2c..2fda618ed5dc 100644 --- a/.github/workflows/rocm-ci.yml +++ b/.github/workflows/rocm-ci.yml @@ -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 }} diff --git a/build/rocm/README.md b/build/rocm/README.md index eb63811df06e..9181f4e9690e 100644 --- a/build/rocm/README.md +++ b/build/rocm/README.md @@ -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) @@ -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 diff --git a/ci/utilities/prepare_rocm_tests.sh b/ci/utilities/prepare_rocm_tests.sh index 0c18f0cbfd49..7a94923062a4 100644 --- a/ci/utilities/prepare_rocm_tests.sh +++ b/ci/utilities/prepare_rocm_tests.sh @@ -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