Merge branch 'THUDM:main' into main

Author: gramesh-amd

.github/workflows/conda-ci.yml

@@ -14,7 +14,7 @@ jobs:
     runs-on: self-hosted
     container:
       image: lmsysorg/sglang:v0.5.0rc0-cu126
-      options: --gpus all --ipc=host --shm-size=16g --ulimit memlock=-1 --ulimit stack=67108864 --memory=0 --memory-swap=0 -v /data/models:/root/models -v /data/datasets:/root/datasets
+      options: --gpus all --ipc=host --shm-size=16g --ulimit memlock=-1 --ulimit stack=67108864 --memory=0 --memory-swap=0 -v /mnt/nvme0n1/models:/root/models -v /mnt/nvme0n1/datasets:/root/datasets
 
     defaults:
       run:

.github/workflows/pr-test.yml

@@ -0,0 +1,57 @@
+name: PR Test
+
+on:
+  # Do not run CI on push to reduce CI time
+  # push:
+  #   branches: [main]
+  pull_request:
+    branches: [main]
+    types: [synchronize, labeled]
+
+concurrency:
+  group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}
+  cancel-in-progress: true
+
+jobs:
+  e2e-test:
+    # TODO may use run-ci label etc
+    if: github.event.pull_request.draft == false
+    runs-on: self-hosted
+    container:
+      image: slimerl/slime:latest
+      options: >
+        --gpus all
+        --ipc=host
+        --shm-size=16g
+        --ulimit memlock=-1
+        --ulimit stack=67108864
+        --memory=0
+        --memory-swap=0
+        -v /mnt/nvme0n1/slime_ci:/data/slime_ci
+        -v /mnt/nvme0n1/models:/root/models
+        -v /mnt/nvme0n1/datasets:/root/datasets
+    strategy:
+      fail-fast: false
+      matrix:
+        info:
+          - {test_file: test_quick_start_glm4_9B.py}
+          - {test_file: test_qwen3_30B_A3B.py}
+          # TODO use deterministic kernel
+    defaults:
+      run:
+        working-directory: ${{ github.workspace }}
+    env:
+      GITHUB_COMMIT_NAME: ${{ github.sha }}_${{ github.event.pull_request.number || 'non-pr' }}
+      WANDB_API_KEY: ${{ secrets.WANDB_API_KEY }}
+
+    steps:
+      - name: Checkout repository
+        uses: actions/checkout@v4
+
+      - name: Install
+        shell: bash
+        run: cd $GITHUB_WORKSPACE && pip install -e .
+
+      - name: Execute
+        shell: bash
+        run: python tests/${{ matrix.info.test_file }}

.github/workflows/pre-commit.yaml

@@ -0,0 +1,41 @@
+name: pre-commit
+
+on:
+  push:
+    branches: [main]
+  pull_request:
+    types: [opened, synchronize, reopened, ready_for_review]
+
+permissions:
+  contents: read
+
+jobs:
+  run-pre-commit:
+    name: Run pre-commit
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout repository
+        uses: actions/checkout@v4
+        with:
+          fetch-depth: 0
+
+      - name: Set up Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: '3.10'
+          cache: 'pip'
+
+      - name: Install pre-commit
+        run: pip install --upgrade pip pre-commit
+
+      - name: Cache pre-commit environments
+        uses: actions/cache@v4
+        with:
+          path: ~/.cache/pre-commit
+          key: pre-commit-${{ runner.os }}-${{ hashFiles('.pre-commit-config.yaml') }}
+          restore-keys: |
+            pre-commit-${{ runner.os }}-
+
+      - name: Run pre-commit on all files
+        run: pre-commit run --all-files --show-diff-on-failure --color=always
+

.github/workflows/pre-commit.yml

@@ -189,4 +189,5 @@ local/
 *.iml
 
 glm/
-_examples_synced/
+_examples_synced/
+.env