From 78c75033642b8c2832fcfb8d483871b65673326b Mon Sep 17 00:00:00 2001 From: Micah Williamson Date: Mon, 8 Dec 2025 20:14:02 -0600 Subject: [PATCH] [ROCm][CI] Skip NVIDIA-Only Prime-RL Test in AMD CI (#29420) Signed-off-by: Micah Williamson --- .buildkite/scripts/run-prime-rl-test.sh | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/.buildkite/scripts/run-prime-rl-test.sh b/.buildkite/scripts/run-prime-rl-test.sh index 5b25c358f..3fb7c82c8 100755 --- a/.buildkite/scripts/run-prime-rl-test.sh +++ b/.buildkite/scripts/run-prime-rl-test.sh @@ -12,6 +12,11 @@ REPO_ROOT="$(cd "${SCRIPT_DIR}/../.." && pwd)" PRIME_RL_REPO="https://github.com/PrimeIntellect-ai/prime-rl.git" PRIME_RL_DIR="${REPO_ROOT}/prime-rl" +if command -v rocm-smi &> /dev/null || command -v rocminfo &> /dev/null; then + echo "AMD GPU detected. Prime-RL currently only supports NVIDIA. Skipping..." + exit 0 +fi + echo "Setting up Prime-RL integration test environment..." # Clean up any existing Prime-RL directory