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
12 changes: 7 additions & 5 deletions .github/workflows/pr-test.yml
Original file line number Diff line number Diff line change
Expand Up @@ -97,7 +97,7 @@ jobs:
- Linux
- ${{ matrix.arch }}
- cpu
timeout-minutes: 300
timeout-minutes: 360
strategy:
fail-fast: false
matrix:
Expand All @@ -108,6 +108,10 @@ jobs:
steps:
- name: Cleanup
run: |
# Stop all Docker containers to free memory
docker stop $(docker ps -q) 2>/dev/null || true
docker rm $(docker ps -aq) 2>/dev/null || true
# Clean workspace and caches
sudo rm -rf ${{ github.workspace }}/* || true
sudo rm -rf ${{ github.workspace }}/.[!.]* || true
rm -rf ~/.cache/flashinfer_jit || true
Expand Down Expand Up @@ -136,7 +140,7 @@ jobs:
needs: setup
if: needs.setup.outputs.skip_build != 'true' && github.event.inputs.skip_gpu != 'true'
runs-on: [self-hosted, Linux, X64, gpu, sm86]
timeout-minutes: 300
timeout-minutes: 360
strategy:
fail-fast: false
matrix:
Expand All @@ -154,7 +158,6 @@ jobs:
sudo rm -rf ${{ github.workspace }}/.[!.]* || true
rm -rf ~/.cache/flashinfer_jit || true
docker system prune -f || true
# Verify GPU is free
nvidia-smi || true

- uses: actions/checkout@v4
Expand All @@ -180,7 +183,7 @@ jobs:
needs: setup
if: needs.setup.outputs.skip_build != 'true' && github.event.inputs.skip_gpu != 'true'
runs-on: [self-hosted, Linux, X64, gpu, sm75]
timeout-minutes: 300
timeout-minutes: 360
env:
DOCKER_IMAGE: flashinfer/flashinfer-ci-cu129:${{ needs.setup.outputs.docker_tag }}
steps:
Expand All @@ -194,7 +197,6 @@ jobs:
sudo rm -rf ${{ github.workspace }}/.[!.]* || true
rm -rf ~/.cache/flashinfer_jit || true
docker system prune -f || true
# Verify GPU is free
nvidia-smi || true

- uses: actions/checkout@v4
Expand Down
2 changes: 1 addition & 1 deletion scripts/task_show_node_info.sh
Original file line number Diff line number Diff line change
Expand Up @@ -40,5 +40,5 @@ ec2_metadata public-hostname
echo "===== RUNNER INFO ====="
df --human-readable
lscpu
free
free -h
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

medium

The pull request description states that this change adds a Docker cleanup step before tests run. However, the only modification in this file is to display the free command output in a human-readable format (free -h). This change improves the readability of the runner info, but it does not implement any Docker cleanup. Please clarify if the Docker cleanup logic is missing or if the PR description needs to be updated to accurately reflect the changes.

nvidia-smi 2>/dev/null || echo "cuda not found"