Skip to content

Commit 78c7503

Browse files
authored
[ROCm][CI] Skip NVIDIA-Only Prime-RL Test in AMD CI (#29420)
Signed-off-by: Micah Williamson <[email protected]>
1 parent e41312a commit 78c7503

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

.buildkite/scripts/run-prime-rl-test.sh

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,11 @@ REPO_ROOT="$(cd "${SCRIPT_DIR}/../.." && pwd)"
1212
PRIME_RL_REPO="https://github.com/PrimeIntellect-ai/prime-rl.git"
1313
PRIME_RL_DIR="${REPO_ROOT}/prime-rl"
1414

15+
if command -v rocm-smi &> /dev/null || command -v rocminfo &> /dev/null; then
16+
echo "AMD GPU detected. Prime-RL currently only supports NVIDIA. Skipping..."
17+
exit 0
18+
fi
19+
1520
echo "Setting up Prime-RL integration test environment..."
1621

1722
# Clean up any existing Prime-RL directory

0 commit comments

Comments
 (0)