Unit Tests #1
Workflow file for this run
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| name: Unit Tests (GPU) | |
| on: | |
| repository_dispatch: | |
| workflow_dispatch: | |
| inputs: | |
| repo: | |
| description: 'GPTQModel repository' | |
| required: true | |
| default: '' | |
| env: | |
| CUDA_DEVICE_ORDER: PCI_BUS_ID | |
| jobs: | |
| build: | |
| runs-on: self-hosted | |
| container: | |
| image: modelcloud/gptqmodel:github-ci-v1 | |
| steps: | |
| - name: Checkout Codes | |
| uses: actions/checkout@v4 | |
| with: | |
| repository: ${{ github.event.inputs.repo }} | |
| - name: Compile | |
| run: python setup.py bdist_wheel | |
| - name: Show dist folder | |
| run: ls -alh dist | |
| - name: Upload to artifact | |
| uses: actions/upload-artifact@v4 | |
| with: | |
| name: dist | |
| path: dist | |
| test: | |
| needs: build | |
| runs-on: self-hosted | |
| container: | |
| image: modelcloud/gptqmodel:github-ci-v1 | |
| strategy: | |
| fail-fast: false | |
| matrix: | |
| test-group: | |
| - gpu_group1 | |
| - gpu_group2 | |
| steps: | |
| - name: Download artifact | |
| uses: actions/download-artifact@v4 | |
| with: | |
| name: dist | |
| path: dist | |
| - name: Show dist folder | |
| run: ls -alh dist | |
| - name: Install wheel | |
| run: | | |
| # install only the last version | |
| pip install dist/*.whl | |
| - name: Find suitable GPU | |
| run: | | |
| suitable_gpu=$(nvidia-smi -L | grep "RTX 4090" | awk -F': ' '{print $1}' | sed 's/GPU //g' | while read gpu_id | |
| do | |
| mem_total=$(nvidia-smi --query-gpu=memory.total --format=csv,noheader,nounits -i $gpu_id) | |
| mem_used=$(nvidia-smi --query-gpu=memory.used --format=csv,noheader,nounits -i $gpu_id) | |
| mem_used_pct=$((100 * mem_used / mem_total)) | |
| if [ $mem_used_pct -lt 2 ]; then # 2 -> 98% free | |
| echo $gpu_id | |
| break | |
| fi | |
| done) | |
| if [ -z "$suitable_gpu" ]; then | |
| echo "No suitable GPU found. Exiting with error." | |
| exit 1 | |
| else | |
| echo "CUDA_VISIBLE_DEVICES=$suitable_gpu" >> $GITHUB_ENV | |
| echo "CUDA_VISIBLE_DEVICES set to $suitable_gpu" | |
| fi | |
| - name: Run tests group 1 | |
| id: test_perplexity | |
| if: matrix.test-group == 'gpu_group1' | |
| run: pytest --durations=0 tests/test_perplexity.py | |
| - name: Run test_lm_head.py | |
| id: test_lm_head | |
| continue-on-error: true | |
| if: matrix.test-group == 'gpu_group2' | |
| run: pytest --durations=0 tests/test_lm_head.py | |
| - name: Run test_q4_exallama.py | |
| id: test_q4_exallama | |
| continue-on-error: true | |
| if: matrix.test-group == 'gpu_group2' | |
| run: pytest --durations=0 tests/test_q4_exallama.py | |
| - name: Run test_q4_exallama_v2.py | |
| id: test_q4_exallama_v2 | |
| continue-on-error: true | |
| if: matrix.test-group == 'gpu_group2' | |
| run: pytest --durations=0 tests/test_q4_exallama_v2.py | |
| - name: Run test_q4_marlin.py | |
| id: test_q4_marlin | |
| continue-on-error: true | |
| if: matrix.test-group == 'gpu_group2' | |
| run: pytest --durations=0 tests/test_q4_marlin.py | |
| - name: Run test_q4_triton.py | |
| id: test_q4_triton | |
| continue-on-error: true | |
| if: matrix.test-group == 'gpu_group2' | |
| run: pytest --durations=0 tests/test_q4_triton.py | |
| - name: Run test_repacking.py | |
| id: test_repacking | |
| continue-on-error: true | |
| if: matrix.test-group == 'gpu_group2' | |
| run: pytest --durations=0 tests/test_repacking.py | |
| - name: Run test_serialization.py | |
| id: test_serialization | |
| continue-on-error: true | |
| if: matrix.test-group == 'gpu_group2' | |
| run: pytest --durations=0 tests/test_serialization.py | |
| - name: Run test_sharded.py | |
| id: test_sharded | |
| continue-on-error: true | |
| if: matrix.test-group == 'gpu_group2' | |
| run: pytest --durations=0 tests/test_sharded.py | |
| - name: Run test_triton.py | |
| id: test_triton | |
| continue-on-error: true | |
| if: matrix.test-group == 'gpu_group2' | |
| run: pytest --durations=0 tests/test_triton.py | |
| - name: Run test_quant_formats.py | |
| id: test_quant_formats | |
| continue-on-error: true | |
| if: matrix.test-group == 'gpu_group2' | |
| run: pytest --durations=0 tests/test_quant_formats.py | |
| - name: Run test_q4_cuda.py | |
| id: test_q4_cuda | |
| continue-on-error: true | |
| if: matrix.test-group == 'gpu_group2' | |
| run: pytest --durations=0 tests/test_q4_cuda.py | |
| - name: Run test_q4_bitblas.py | |
| id: test_q4_bitblas | |
| continue-on-error: true | |
| if: matrix.test-group == 'gpu_group2' | |
| run: pytest --durations=0 tests/test_q4_bitblas.py | |
| - name: Print results | |
| shell: bash | |
| if: matrix.test-group == 'gpu_group2' | |
| run: | | |
| declare -A step_outcomes | |
| step_outcomes=( | |
| [test_lm_head]="${{ steps.test_lm_head.outcome }}" | |
| [test_q4_exallama]="${{ steps.test_q4_exallama.outcome }}" | |
| [test_q4_exallama_v2]="${{ steps.test_q4_exallama_v2.outcome }}" | |
| [test_q4_marlin]="${{ steps.test_q4_marlin.outcome }}" | |
| [test_q4_triton]="${{ steps.test_q4_triton.outcome }}" | |
| [test_repacking]="${{ steps.test_repacking.outcome }}" | |
| [test_serialization]="${{ steps.test_serialization.outcome }}" | |
| [test_sharded]="${{ steps.test_sharded.outcome }}" | |
| [test_triton]="${{ steps.test_triton.outcome }}" | |
| [test_quant_formats]="${{ steps.test_quant_formats.outcome }}" | |
| [test_q4_cuda]="${{ steps.test_q4_cuda.outcome }}" | |
| [test_q4_bitblas]="${{ steps.test_q4_bitblas.outcome }}" | |
| ) | |
| max_length=0 | |
| for step in "${!step_outcomes[@]}"; do | |
| length=${#step} | |
| if [[ $length -gt $max_length ]]; then | |
| max_length=$length | |
| fi | |
| done | |
| error_occurred=0 | |
| for step in "${!step_outcomes[@]}"; do | |
| outcome="${step_outcomes[$step]}" | |
| if [ "$outcome" == "success" ]; then | |
| printf "\e[32m%-*s Result: %s\e[0m\n" $((max_length + 4)) "$step" "$outcome" | |
| else | |
| printf "\e[31m%-*s Result: %s\e[0m\n" $((max_length + 4)) "$step" "$outcome" | |
| error_occurred=1 | |
| fi | |
| done | |
| if [ $error_occurred -eq 1 ]; then | |
| exit 1 | |
| fi |