Unverified Commit 121ea5a2 authored by Divin Honnappa's avatar Divin Honnappa Committed by GitHub
Browse files

Removed GPU state confirmation and cleanup steps. (#38238)


Signed-off-by: default avatarDivin Honnappa <divin.honnappa@amd.com>
parent ab79863e
...@@ -35,23 +35,6 @@ export PYTHONPATH=".." ...@@ -35,23 +35,6 @@ export PYTHONPATH=".."
# Helper Functions # Helper Functions
############################################################################### ###############################################################################
wait_for_clean_gpus() {
local timeout=${1:-300}
local start=$SECONDS
echo "--- Waiting for clean GPU state (timeout: ${timeout}s)"
while true; do
if grep -q clean /opt/amdgpu/etc/gpu_state; then
echo "GPUs state is \"clean\""
return
fi
if (( SECONDS - start >= timeout )); then
echo "Error: GPUs did not reach clean state within ${timeout}s" >&2
exit 1
fi
sleep 3
done
}
cleanup_docker() { cleanup_docker() {
# Get Docker's root directory # Get Docker's root directory
docker_root=$(docker info -f '{{.DockerRootDir}}') docker_root=$(docker info -f '{{.DockerRootDir}}')
...@@ -365,19 +348,12 @@ apply_rocm_test_overrides() { ...@@ -365,19 +348,12 @@ apply_rocm_test_overrides() {
############################################################################### ###############################################################################
# --- GPU initialization --- # --- GPU initialization ---
echo "--- Confirming Clean Initial State"
wait_for_clean_gpus
echo "--- ROCm info" echo "--- ROCm info"
rocminfo rocminfo
# --- Docker housekeeping --- # --- Docker housekeeping ---
cleanup_docker cleanup_docker
echo "--- Resetting GPUs"
echo "reset" > /opt/amdgpu/etc/gpu_state
wait_for_clean_gpus
# --- Pull test image --- # --- Pull test image ---
echo "--- Pulling container" echo "--- Pulling container"
image_name="rocm/vllm-ci:${BUILDKITE_COMMIT}" image_name="rocm/vllm-ci:${BUILDKITE_COMMIT}"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment