Compare commits
11 Commits
fix/48-mar
...
step35/98-
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
e20439b544 | ||
| 7797b9b4c8 | |||
| 0338cf940a | |||
| f3f796fa64 | |||
| 6ab98d65f5 | |||
| c4293f0d31 | |||
| 88a5c48402 | |||
| 3ff52f02b2 | |||
| 8475539070 | |||
|
|
f0f117cdd3 | ||
|
|
a537511652 |
@@ -18,6 +18,13 @@ jobs:
|
|||||||
find . -name '*.py' | grep -v llama-cpp-fork | xargs -r python3 -m py_compile
|
find . -name '*.py' | grep -v llama-cpp-fork | xargs -r python3 -m py_compile
|
||||||
find . -name '*.sh' | xargs -r bash -n
|
find . -name '*.sh' | xargs -r bash -n
|
||||||
echo "PASS: All files parse"
|
echo "PASS: All files parse"
|
||||||
|
- name: Build standalone CMake target
|
||||||
|
run: |
|
||||||
|
cmake -S . -B build -DTURBOQUANT_BUILD_TESTS=ON
|
||||||
|
cmake --build build -j$(nproc)
|
||||||
|
- name: Run tests
|
||||||
|
run: |
|
||||||
|
ctest --test-dir build --output-on-failure
|
||||||
- name: Secret scan
|
- name: Secret scan
|
||||||
run: |
|
run: |
|
||||||
if grep -rE 'sk-or-|sk-ant-|ghp_|AKIA' . --include='*.yml' --include='*.py' --include='*.sh' 2>/dev/null | grep -v .gitea | grep -v llama-cpp-fork; then exit 1; fi
|
if grep -rE 'sk-or-|sk-ant-|ghp_|AKIA' . --include='*.yml' --include='*.py' --include='*.sh' 2>/dev/null | grep -v .gitea | grep -v llama-cpp-fork; then exit 1; fi
|
||||||
|
|||||||
19
ansible/README.md
Normal file
19
ansible/README.md
Normal file
@@ -0,0 +1,19 @@
|
|||||||
|
# TurboQuant Ansible Deployment
|
||||||
|
|
||||||
|
Deploy TurboQuant-compressed Gemma 4 inference across fleet nodes.
|
||||||
|
|
||||||
|
## Quick Start
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# Copy and edit inventory
|
||||||
|
cp ansible/inventory.ini.example ansible/inventory.ini
|
||||||
|
|
||||||
|
# Deploy to all nodes
|
||||||
|
ansible-playbook -i ansible/inventory.ini ansible/deploy_turboquant.yml
|
||||||
|
|
||||||
|
# Run health check
|
||||||
|
ansible -i ansible/inventory.ini all -m shell -a "sudo /opt/turboquant/health_check.sh"
|
||||||
|
|
||||||
|
# Run integration test
|
||||||
|
ansible -i ansible/inventory.ini all -m shell -a "curl -s http://localhost:8081/v1/chat/completions -d '{\"model\":\"gemma-4\",\"messages\":[{\"role\":\"user\",\"content\":\"Hello\"}]}'"
|
||||||
|
```
|
||||||
69
ansible/deploy_turboquant.yml
Normal file
69
ansible/deploy_turboquant.yml
Normal file
@@ -0,0 +1,69 @@
|
|||||||
|
---
|
||||||
|
# deploy_turboquant.yml — Deploy TurboQuant across fleet nodes
|
||||||
|
# Usage: ansible-playbook -i ansible/inventory.ini ansible/deploy_turboquant.yml
|
||||||
|
|
||||||
|
- name: Deploy TurboQuant to Mac (local)
|
||||||
|
hosts: mac
|
||||||
|
become: yes
|
||||||
|
gather_facts: yes
|
||||||
|
|
||||||
|
vars:
|
||||||
|
turboquant_user: "turboquant"
|
||||||
|
turboquant_install_dir: "/opt/turboquant"
|
||||||
|
turboquant_service_name: "turboquant"
|
||||||
|
turboquant_port: 8081
|
||||||
|
turboquant_host: "0.0.0.0"
|
||||||
|
turboquant_context: 131072
|
||||||
|
turboquant_model: "gemma-4"
|
||||||
|
turboquant_model_file: "gemma-4-26B-A4B.gguf"
|
||||||
|
turboquant_kv_type: "turbo4"
|
||||||
|
turboquant_layer_adaptive: 7
|
||||||
|
node_preset: "turboquant_k8v4"
|
||||||
|
node_hardware: "M1-16GB"
|
||||||
|
|
||||||
|
roles:
|
||||||
|
- turboquant-deploy
|
||||||
|
|
||||||
|
- name: Deploy TurboQuant to Allegro VPS
|
||||||
|
hosts: allegro
|
||||||
|
become: yes
|
||||||
|
gather_facts: yes
|
||||||
|
|
||||||
|
vars:
|
||||||
|
turboquant_user: "turboquant"
|
||||||
|
turboquant_install_dir: "/opt/turboquant"
|
||||||
|
turboquant_service_name: "turboquant"
|
||||||
|
turboquant_port: 8081
|
||||||
|
turboquant_host: "0.0.0.0"
|
||||||
|
turboquant_context: 65536
|
||||||
|
turboquant_model: "gemma-4-E4B"
|
||||||
|
turboquant_model_file: "gemma-4-E4B.gguf"
|
||||||
|
turboquant_kv_type: "q4_0"
|
||||||
|
turboquant_layer_adaptive: 0
|
||||||
|
node_preset: "turboquant_4bit_nc"
|
||||||
|
node_hardware: "VPS-2c8g"
|
||||||
|
|
||||||
|
roles:
|
||||||
|
- turboquant-deploy
|
||||||
|
|
||||||
|
- name: Deploy TurboQuant to Ezra VPS
|
||||||
|
hosts: ezra
|
||||||
|
become: yes
|
||||||
|
gather_facts: yes
|
||||||
|
|
||||||
|
vars:
|
||||||
|
turboquant_user: "turboquant"
|
||||||
|
turboquant_install_dir: "/opt/turboquant"
|
||||||
|
turboquant_service_name: "turboquant"
|
||||||
|
turboquant_port: 8081
|
||||||
|
turboquant_host: "0.0.0.0"
|
||||||
|
turboquant_context: 65536
|
||||||
|
turboquant_model: "gemma-4-E4B"
|
||||||
|
turboquant_model_file: "gemma-4-E4B.gguf"
|
||||||
|
turboquant_kv_type: "q4_0"
|
||||||
|
turboquant_layer_adaptive: 0
|
||||||
|
node_preset: "turboquant_4bit_nc"
|
||||||
|
node_hardware: "VPS-2c8g"
|
||||||
|
|
||||||
|
roles:
|
||||||
|
- turboquant-deploy
|
||||||
23
ansible/health_check.sh
Executable file
23
ansible/health_check.sh
Executable file
@@ -0,0 +1,23 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
# Health check for TurboQuant llama-server / vLLM deployment
|
||||||
|
|
||||||
|
set -e
|
||||||
|
|
||||||
|
PORT="${TURBOQUANT_PORT:-8081}"
|
||||||
|
ENDPOINT="${TURBOQUANT_ENDPOINT:-http://localhost:${PORT}/v1/models}"
|
||||||
|
|
||||||
|
echo "Checking TurboQuant server health at ${ENDPOINT}..."
|
||||||
|
|
||||||
|
if command -v curl &> /dev/null; then
|
||||||
|
response=$(curl -s -o /dev/null -w "%{http_code}" "${ENDPOINT}" --max-time 10)
|
||||||
|
if [ "${response}" = "200" ]; then
|
||||||
|
echo "✅ Server healthy — HTTP ${response}"
|
||||||
|
exit 0
|
||||||
|
else
|
||||||
|
echo "❌ Server unhealthy — HTTP ${response}"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
else
|
||||||
|
echo "curl not found; cannot perform health check"
|
||||||
|
exit 2
|
||||||
|
fi
|
||||||
22
ansible/inventory.ini.example
Normal file
22
ansible/inventory.ini.example
Normal file
@@ -0,0 +1,22 @@
|
|||||||
|
# Ansible inventory for TurboQuant fleet deployment
|
||||||
|
# Edit this file and save as ansible/inventory.ini before running
|
||||||
|
|
||||||
|
[mac]
|
||||||
|
# Local MacBook — runs llama-server with Metal + TurboQuant
|
||||||
|
timmy-mac ansible_host=localhost ansible_connection=local
|
||||||
|
|
||||||
|
[allegro]
|
||||||
|
# Allegro VPS — Debian, runs llama-server or vLLM with GGUF q4_0
|
||||||
|
allegro-primary ansible_host=167.99.126.228 ansible_user=root
|
||||||
|
|
||||||
|
[ezra]
|
||||||
|
# Ezra VPS — Ubuntu, runs llama-server or vLLM
|
||||||
|
ezra-primary ansible_host=143.198.27.163 ansible_user=root ansible_connection=local
|
||||||
|
|
||||||
|
[turbonodes:children]
|
||||||
|
mac
|
||||||
|
allegro
|
||||||
|
ezra
|
||||||
|
|
||||||
|
[turbonodes:vars]
|
||||||
|
ansible_python_interpreter=/usr/bin/python3
|
||||||
68
ansible/roles/turboquant-deploy/tasks/darwin.yml
Normal file
68
ansible/roles/turboquant-deploy/tasks/darwin.yml
Normal file
@@ -0,0 +1,68 @@
|
|||||||
|
---
|
||||||
|
# macOS deployment — builds llama.cpp with Metal + TurboQuant
|
||||||
|
|
||||||
|
- name: Ensure Xcode command line tools are installed
|
||||||
|
command: xcode-select -p
|
||||||
|
register: xcode_check
|
||||||
|
changed_when: false
|
||||||
|
failed_when: false
|
||||||
|
when: ansible_os_family == "Darwin"
|
||||||
|
tags: [turboquant, darwin]
|
||||||
|
|
||||||
|
- name: Install Xcode CLI tools if missing (macOS)
|
||||||
|
shell: xcode-select --install
|
||||||
|
when: ansible_os_family == "Darwin" and xcode_check.rc != 0
|
||||||
|
tags: [turboquant, darwin]
|
||||||
|
|
||||||
|
- name: Check for Git
|
||||||
|
command: which git
|
||||||
|
register: git_check
|
||||||
|
when: ansible_os_family == "Darwin"
|
||||||
|
tags: [turboquant, deps]
|
||||||
|
|
||||||
|
- name: Clone llama.cpp TurboQuant fork
|
||||||
|
git:
|
||||||
|
repo: "https://github.com/TheTom/llama-cpp-turboquant.git"
|
||||||
|
dest: "{{ turboquant_install_dir }}/llama.cpp"
|
||||||
|
version: "feature/turboquant-kv-cache"
|
||||||
|
force: yes
|
||||||
|
when: ansible_os_family == "Darwin"
|
||||||
|
tags: [turboquant, source]
|
||||||
|
|
||||||
|
- name: Build llama.cpp with Metal + TurboQuant
|
||||||
|
shell: |
|
||||||
|
cd {{ turboquant_install_dir }}/llama.cpp
|
||||||
|
cmake -B build -DCMAKE_BUILD_TYPE=Release -DGGML_METAL=ON
|
||||||
|
cmake --build build -j$(sysctl -n hw.ncpu)
|
||||||
|
args:
|
||||||
|
creates: "{{ turboquant_install_dir }}/llama.cpp/build/bin/llama-server"
|
||||||
|
when: ansible_os_family == "Darwin"
|
||||||
|
tags: [turboquant, build]
|
||||||
|
|
||||||
|
- name: Create models directory
|
||||||
|
file:
|
||||||
|
path: "{{ turboquant_install_dir }}/models"
|
||||||
|
state: directory
|
||||||
|
mode: '0755'
|
||||||
|
when: ansible_os_family == "Darwin"
|
||||||
|
tags: [turboquant, deploy]
|
||||||
|
|
||||||
|
- name: Deploy health check script
|
||||||
|
copy:
|
||||||
|
src: "../../health_check.sh"
|
||||||
|
dest: "{{ turboquant_install_dir }}/health_check.sh"
|
||||||
|
mode: '0755'
|
||||||
|
when: ansible_os_family == "Darwin"
|
||||||
|
tags: [turboquant, deploy]
|
||||||
|
|
||||||
|
- name: Print macOS manual start instructions
|
||||||
|
debug:
|
||||||
|
msg: |
|
||||||
|
Mac deployment complete. To start the server manually:
|
||||||
|
export TURBO_LAYER_ADAPTIVE={{ turboquant_layer_adaptive }}
|
||||||
|
sudo -u {{ turboquant_user }} {{ turboquant_install_dir }}/llama.cpp/build/bin/llama-server \
|
||||||
|
-m {{ turboquant_install_dir }}/models/{{ turboquant_model_file }} \
|
||||||
|
--host {{ turboquant_host }} --port {{ turboquant_port }} \
|
||||||
|
-c {{ turboquant_context }} -ctk {{ turboquant_kv_type }} -ctv {{ turboquant_kv_type }}
|
||||||
|
when: ansible_os_family == "Darwin"
|
||||||
|
tags: [turboquant, deploy]
|
||||||
92
ansible/roles/turboquant-deploy/tasks/debian.yml
Normal file
92
ansible/roles/turboquant-deploy/tasks/debian.yml
Normal file
@@ -0,0 +1,92 @@
|
|||||||
|
---
|
||||||
|
# Debian/Ubuntu deployment — installs llama.cpp with TurboQuant, uses systemd
|
||||||
|
|
||||||
|
- name: Update apt cache
|
||||||
|
apt:
|
||||||
|
update_cache: yes
|
||||||
|
cache_valid_time: 3600
|
||||||
|
tags: [turboquant, deps]
|
||||||
|
|
||||||
|
- name: Install build dependencies
|
||||||
|
apt:
|
||||||
|
name:
|
||||||
|
- build-essential
|
||||||
|
- cmake
|
||||||
|
- git
|
||||||
|
- curl
|
||||||
|
- python3
|
||||||
|
- python3-pip
|
||||||
|
- python3-venv
|
||||||
|
state: present
|
||||||
|
tags: [turboquant, deps]
|
||||||
|
|
||||||
|
- name: Create turboquant user
|
||||||
|
user:
|
||||||
|
name: "{{ turboquant_user }}"
|
||||||
|
system: yes
|
||||||
|
shell: /usr/sbin/nologin
|
||||||
|
create_home: no
|
||||||
|
tags: [turboquant, prereq]
|
||||||
|
|
||||||
|
- name: Create install directory
|
||||||
|
file:
|
||||||
|
path: "{{ turboquant_install_dir }}"
|
||||||
|
state: directory
|
||||||
|
mode: '0755'
|
||||||
|
owner: "{{ turboquant_user }}"
|
||||||
|
group: "{{ turboquant_user }}"
|
||||||
|
tags: [turboquant, prereq]
|
||||||
|
|
||||||
|
- name: Clone llama.cpp TurboQuant fork
|
||||||
|
git:
|
||||||
|
repo: "https://github.com/TheTom/llama-cpp-turboquant.git"
|
||||||
|
dest: "{{ turboquant_install_dir }}/llama.cpp"
|
||||||
|
version: "feature/turboquant-kv-cache"
|
||||||
|
force: yes
|
||||||
|
tags: [turboquant, source]
|
||||||
|
|
||||||
|
- name: Build llama.cpp with TurboQuant
|
||||||
|
shell: |
|
||||||
|
cd {{ turboquant_install_dir }}/llama.cpp
|
||||||
|
cmake -B build -DCMAKE_BUILD_TYPE=Release
|
||||||
|
cmake --build build -j$(nproc)
|
||||||
|
args:
|
||||||
|
creates: "{{ turboquant_install_dir }}/llama.cpp/build/bin/llama-server"
|
||||||
|
tags: [turboquant, build]
|
||||||
|
|
||||||
|
- name: Create models directory
|
||||||
|
file:
|
||||||
|
path: "{{ turboquant_install_dir }}/models"
|
||||||
|
state: directory
|
||||||
|
mode: '0755'
|
||||||
|
owner: "{{ turboquant_user }}"
|
||||||
|
group: "{{ turboquant_user }}"
|
||||||
|
tags: [turboquant, deploy]
|
||||||
|
|
||||||
|
- name: Deploy systemd service unit
|
||||||
|
template:
|
||||||
|
src: turboquant.service.j2
|
||||||
|
dest: /etc/systemd/system/{{ turboquant_service_name }}.service
|
||||||
|
mode: '0644'
|
||||||
|
tags: [turboquant, service]
|
||||||
|
|
||||||
|
- name: Reload systemd daemon
|
||||||
|
systemd:
|
||||||
|
daemon_reload: yes
|
||||||
|
tags: [turboquant, service]
|
||||||
|
|
||||||
|
- name: Enable and start TurboQuant service
|
||||||
|
systemd:
|
||||||
|
name: "{{ turboquant_service_name }}"
|
||||||
|
state: started
|
||||||
|
enabled: yes
|
||||||
|
tags: [turboquant, service]
|
||||||
|
|
||||||
|
- name: Deploy health check script
|
||||||
|
copy:
|
||||||
|
src: "../../health_check.sh"
|
||||||
|
dest: "{{ turboquant_install_dir }}/health_check.sh"
|
||||||
|
mode: '0755'
|
||||||
|
owner: "{{ turboquant_user }}"
|
||||||
|
group: "{{ turboquant_user }}"
|
||||||
|
tags: [turboquant, deploy]
|
||||||
45
ansible/roles/turboquant-deploy/tasks/integration_test.yml
Normal file
45
ansible/roles/turboquant-deploy/tasks/integration_test.yml
Normal file
@@ -0,0 +1,45 @@
|
|||||||
|
---
|
||||||
|
# Integration test — verify server responds to a simple query
|
||||||
|
|
||||||
|
- name: Wait for service to be ready (HTTP 200 on /v1/models)
|
||||||
|
uri:
|
||||||
|
url: "http://localhost:{{ turboquant_port }}/v1/models"
|
||||||
|
method: GET
|
||||||
|
status_code: 200
|
||||||
|
register: svc_ready
|
||||||
|
retries: 12
|
||||||
|
delay: 5
|
||||||
|
until: svc_ready.status == 200
|
||||||
|
when: ansible_os_family != "Darwin" # skip on mac for now; service starts manually
|
||||||
|
tags: [turboquant, healthcheck]
|
||||||
|
|
||||||
|
- name: Run integration test — simple query
|
||||||
|
uri:
|
||||||
|
url: "http://localhost:{{ turboquant_port }}/v1/chat/completions"
|
||||||
|
method: POST
|
||||||
|
body_format: json
|
||||||
|
body:
|
||||||
|
model: "{{ turboquant_model }}"
|
||||||
|
messages:
|
||||||
|
- role: "user"
|
||||||
|
content: "Test: 2+2 equals what? Answer with only the number."
|
||||||
|
max_tokens: 5
|
||||||
|
temperature: 0.0
|
||||||
|
return_content: yes
|
||||||
|
register: completion
|
||||||
|
when: ansible_os_family != "Darwin"
|
||||||
|
tags: [turboquant, test]
|
||||||
|
|
||||||
|
- name: Verify response contains expected answer
|
||||||
|
assert:
|
||||||
|
that:
|
||||||
|
- "'4' in (completion.content | default(''))"
|
||||||
|
- completion.status == 200
|
||||||
|
when: ansible_os_family != "Darwin"
|
||||||
|
tags: [turboquant, test]
|
||||||
|
|
||||||
|
- name: Log integration result
|
||||||
|
debug:
|
||||||
|
msg: "Integration test passed — TurboQuant server responded correctly"
|
||||||
|
when: ansible_os_family != "Darwin"
|
||||||
|
tags: [turboquant, test]
|
||||||
17
ansible/roles/turboquant-deploy/tasks/main.yml
Normal file
17
ansible/roles/turboquant-deploy/tasks/main.yml
Normal file
@@ -0,0 +1,17 @@
|
|||||||
|
---
|
||||||
|
# Main entry point — common setup followed by OS-specific tasks
|
||||||
|
|
||||||
|
- name: Ensure install directory exists (common)
|
||||||
|
file:
|
||||||
|
path: "{{ turboquant_install_dir }}"
|
||||||
|
state: directory
|
||||||
|
mode: '0755'
|
||||||
|
tags: [turboquant, prereq]
|
||||||
|
|
||||||
|
- name: Include OS-specific tasks
|
||||||
|
include_tasks: "{{ ansible_os_family | lower }}.yml"
|
||||||
|
tags: [turboquant, deploy]
|
||||||
|
|
||||||
|
- name: Run post-deploy integration tests
|
||||||
|
include_tasks: integration_test.yml
|
||||||
|
tags: [turboquant, test]
|
||||||
@@ -0,0 +1,25 @@
|
|||||||
|
---
|
||||||
|
# TurboQuant Server Configuration
|
||||||
|
# Auto-generated by Ansible — node: {{ ansible_host | default('localhost') }}
|
||||||
|
|
||||||
|
server:
|
||||||
|
host: "{{ turboquant_host }}"
|
||||||
|
port: {{ turboquant_port }}
|
||||||
|
model: "{{ turboquant_model }}"
|
||||||
|
model_file: "{{ turboquant_model_file }}"
|
||||||
|
base_url: "http://localhost:{{ turboquant_port }}/v1"
|
||||||
|
|
||||||
|
turboquant:
|
||||||
|
enabled: true
|
||||||
|
preset: "{{ node_preset }}"
|
||||||
|
kv_type: "{{ turboquant_kv_type }}"
|
||||||
|
layer_adaptive_mode: {{ turboquant_layer_adaptive }}
|
||||||
|
|
||||||
|
performance:
|
||||||
|
max_context: {{ turboquant_context }}
|
||||||
|
threads: {{ ansible_processor_vcpus | default(2) }}
|
||||||
|
|
||||||
|
deployment:
|
||||||
|
install_dir: "{{ turboquant_install_dir }}"
|
||||||
|
service_name: "{{ turboquant_service_name }}"
|
||||||
|
node_hardware: "{{ node_hardware }}"
|
||||||
@@ -0,0 +1,25 @@
|
|||||||
|
[Unit]
|
||||||
|
Description=TurboQuant {{ turboquant_model }} Inference Server
|
||||||
|
After=network.target
|
||||||
|
|
||||||
|
[Service]
|
||||||
|
Type=simple
|
||||||
|
User={{ turboquant_user }}
|
||||||
|
Group={{ turboquant_user }}
|
||||||
|
WorkingDirectory={{ turboquant_install_dir }}
|
||||||
|
Environment="TURBO_LAYER_ADAPTIVE={{ turboquant_layer_adaptive }}"
|
||||||
|
ExecStart={{ turboquant_install_dir }}/llama-server \
|
||||||
|
-m {{ turboquant_install_dir }}/models/{{ turboquant_model_file }} \
|
||||||
|
--host {{ turboquant_host }} \
|
||||||
|
--port {{ turboquant_port }} \
|
||||||
|
-c {{ turboquant_context }} \
|
||||||
|
-ctk {{ turboquant_kv_type }} -ctv {{ turboquant_kv_type }} \
|
||||||
|
--threads {{ ansible_processor_vcpus | default(2) }}
|
||||||
|
|
||||||
|
Restart=always
|
||||||
|
RestartSec=5
|
||||||
|
StandardOutput=journal
|
||||||
|
StandardError=journal
|
||||||
|
|
||||||
|
[Install]
|
||||||
|
WantedBy=multi-user.target
|
||||||
@@ -385,7 +385,7 @@ Step 7: If pass → production. If fail → drop to turbo3 or adjust per-layer p
|
|||||||
|
|
||||||
---
|
---
|
||||||
|
|
||||||
*Repo: http://143.198.27.163:3000/Timmy_Foundation/turboquant*
|
*Repo: https://forge.alexanderwhitestone.com/Timmy_Foundation/turboquant*
|
||||||
*Build: /tmp/llama-cpp-turboquant/build/bin/ (all binaries)*
|
*Build: /tmp/llama-cpp-turboquant/build/bin/ (all binaries)*
|
||||||
*Branch: feature/turboquant-kv-cache*
|
*Branch: feature/turboquant-kv-cache*
|
||||||
|
|
||||||
|
|||||||
@@ -1,5 +1,29 @@
|
|||||||
"""Phase 19: Hardware-Aware Inference Optimization.
|
"""Backward-compatible shim for hardware-aware quantization selection.
|
||||||
Part of the TurboQuant suite for local inference excellence.
|
|
||||||
|
The original Phase 19 placeholder `hardware_optimizer.py` never shipped real
|
||||||
|
logic. The canonical implementation now lives in `evolution.quant_selector`.
|
||||||
|
This shim preserves the legacy import path for any downstream callers while
|
||||||
|
making `quant_selector.py` the single source of truth.
|
||||||
"""
|
"""
|
||||||
import logging
|
|
||||||
# ... (rest of the code)
|
from evolution.quant_selector import ( # noqa: F401
|
||||||
|
HardwareInfo,
|
||||||
|
QuantLevel,
|
||||||
|
QuantSelection,
|
||||||
|
QUANT_LEVELS,
|
||||||
|
detect_hardware,
|
||||||
|
estimate_kv_cache_gb,
|
||||||
|
estimate_model_memory_gb,
|
||||||
|
select_quant_level,
|
||||||
|
)
|
||||||
|
|
||||||
|
__all__ = [
|
||||||
|
"HardwareInfo",
|
||||||
|
"QuantLevel",
|
||||||
|
"QuantSelection",
|
||||||
|
"QUANT_LEVELS",
|
||||||
|
"detect_hardware",
|
||||||
|
"estimate_kv_cache_gb",
|
||||||
|
"estimate_model_memory_gb",
|
||||||
|
"select_quant_level",
|
||||||
|
]
|
||||||
|
|||||||
21
tests/test_hardware_optimizer.py
Normal file
21
tests/test_hardware_optimizer.py
Normal file
@@ -0,0 +1,21 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
"""Tests for hardware_optimizer compatibility shim."""
|
||||||
|
|
||||||
|
import os
|
||||||
|
import sys
|
||||||
|
|
||||||
|
sys.path.insert(0, os.path.dirname(os.path.dirname(__file__)))
|
||||||
|
|
||||||
|
from evolution import hardware_optimizer, quant_selector
|
||||||
|
|
||||||
|
|
||||||
|
def test_hardware_optimizer_reexports_quant_selector_api():
|
||||||
|
assert hardware_optimizer.select_quant_level is quant_selector.select_quant_level
|
||||||
|
assert hardware_optimizer.detect_hardware is quant_selector.detect_hardware
|
||||||
|
assert hardware_optimizer.HardwareInfo is quant_selector.HardwareInfo
|
||||||
|
assert hardware_optimizer.QuantSelection is quant_selector.QuantSelection
|
||||||
|
|
||||||
|
|
||||||
|
def test_hardware_optimizer_exports_quant_level_definitions():
|
||||||
|
assert hardware_optimizer.QUANT_LEVELS is quant_selector.QUANT_LEVELS
|
||||||
|
assert hardware_optimizer.QuantLevel is quant_selector.QuantLevel
|
||||||
@@ -20,9 +20,35 @@ from evolution.quant_selector import (
|
|||||||
|
|
||||||
class TestQuantLevels:
|
class TestQuantLevels:
|
||||||
def test_levels_ordered_by_quality(self):
|
def test_levels_ordered_by_quality(self):
|
||||||
"""Levels should be ordered from best quality to most aggressive."""
|
"""TurboQuant levels should be ordered from best quality to most aggressive.
|
||||||
for i in range(len(QUANT_LEVELS) - 1):
|
|
||||||
assert QUANT_LEVELS[i].bits_per_channel > QUANT_LEVELS[i + 1].bits_per_channel
|
The quality ordering invariant for TurboQuant levels is monotonically
|
||||||
|
increasing compression_ratio (more aggressive = more compression).
|
||||||
|
Non-TurboQuant fallbacks (e.g. q4_0) are placed after all TurboQuant
|
||||||
|
levels and may have any compression ratio — they exist as safe defaults,
|
||||||
|
not as part of the quality progression.
|
||||||
|
"""
|
||||||
|
turbo_quant_names = {"turbo4", "turbo3", "turbo2"}
|
||||||
|
turbo_levels = [l for l in QUANT_LEVELS if l.name in turbo_quant_names]
|
||||||
|
for i in range(len(turbo_levels) - 1):
|
||||||
|
assert turbo_levels[i].compression_ratio <= turbo_levels[i + 1].compression_ratio, (
|
||||||
|
f"TurboQuant {turbo_levels[i].name} (compression={turbo_levels[i].compression_ratio}x) "
|
||||||
|
f"should have <= compression than {turbo_levels[i+1].name} "
|
||||||
|
f"(compression={turbo_levels[i+1].compression_ratio}x)"
|
||||||
|
)
|
||||||
|
|
||||||
|
def test_fallback_quant_is_last(self):
|
||||||
|
"""Non-TurboQuant fallbacks (e.g. q4_0) should be at the end of the list."""
|
||||||
|
turbo_quant_names = {"turbo4", "turbo3", "turbo2"}
|
||||||
|
found_fallback = False
|
||||||
|
for level in QUANT_LEVELS:
|
||||||
|
if level.name not in turbo_quant_names:
|
||||||
|
found_fallback = True
|
||||||
|
elif found_fallback:
|
||||||
|
pytest.fail(
|
||||||
|
f"TurboQuant level '{level.name}' appears after a fallback level. "
|
||||||
|
f"All TurboQuant levels must precede fallbacks."
|
||||||
|
)
|
||||||
|
|
||||||
def test_all_levels_have_required_fields(self):
|
def test_all_levels_have_required_fields(self):
|
||||||
for level in QUANT_LEVELS:
|
for level in QUANT_LEVELS:
|
||||||
|
|||||||
83
tests/test_smoke_workflow.py
Normal file
83
tests/test_smoke_workflow.py
Normal file
@@ -0,0 +1,83 @@
|
|||||||
|
"""Tests for smoke workflow CI configuration.
|
||||||
|
|
||||||
|
Validates that the GitHub Actions / Gitea Actions smoke workflow
|
||||||
|
actually runs the standalone CMake build and test suite, not just
|
||||||
|
parse checks.
|
||||||
|
"""
|
||||||
|
|
||||||
|
from pathlib import Path
|
||||||
|
|
||||||
|
import yaml
|
||||||
|
|
||||||
|
import pytest
|
||||||
|
|
||||||
|
|
||||||
|
WORKFLOW_PATH = Path(".gitea/workflows/smoke.yml")
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture
|
||||||
|
def workflow():
|
||||||
|
"""Load and parse the smoke workflow YAML."""
|
||||||
|
content = WORKFLOW_PATH.read_text(encoding="utf-8")
|
||||||
|
return yaml.safe_load(content)
|
||||||
|
|
||||||
|
|
||||||
|
def test_smoke_workflow_exists():
|
||||||
|
"""Smoke workflow file must exist."""
|
||||||
|
assert WORKFLOW_PATH.exists(), f"Missing {WORKFLOW_PATH}"
|
||||||
|
|
||||||
|
|
||||||
|
def test_smoke_has_cmake_configure_step(workflow):
|
||||||
|
"""Smoke workflow must configure the CMake project with tests enabled."""
|
||||||
|
steps = workflow["jobs"]["smoke"]["steps"]
|
||||||
|
cmake_found = False
|
||||||
|
for step in steps:
|
||||||
|
run = step.get("run", "")
|
||||||
|
if "cmake -S . -B build" in run and "TURBOQUANT_BUILD_TESTS=ON" in run:
|
||||||
|
cmake_found = True
|
||||||
|
break
|
||||||
|
assert cmake_found, (
|
||||||
|
"Smoke workflow missing cmake configure step with TURBOQUANT_BUILD_TESTS=ON"
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def test_smoke_has_cmake_build_step(workflow):
|
||||||
|
"""Smoke workflow must build the CMake project."""
|
||||||
|
steps = workflow["jobs"]["smoke"]["steps"]
|
||||||
|
build_found = False
|
||||||
|
for step in steps:
|
||||||
|
run = step.get("run", "")
|
||||||
|
if "cmake --build build" in run:
|
||||||
|
build_found = True
|
||||||
|
break
|
||||||
|
assert build_found, "Smoke workflow missing cmake --build step"
|
||||||
|
|
||||||
|
|
||||||
|
def test_smoke_has_ctest_step(workflow):
|
||||||
|
"""Smoke workflow must run ctest."""
|
||||||
|
steps = workflow["jobs"]["smoke"]["steps"]
|
||||||
|
ctest_found = False
|
||||||
|
for step in steps:
|
||||||
|
run = step.get("run", "")
|
||||||
|
if "ctest" in run and "output-on-failure" in run:
|
||||||
|
ctest_found = True
|
||||||
|
break
|
||||||
|
assert ctest_found, "Smoke workflow missing ctest --output-on-failure step"
|
||||||
|
|
||||||
|
|
||||||
|
def test_smoke_build_before_secret_scan(workflow):
|
||||||
|
"""Build and test steps must run before secret scan (fail fast on build errors)."""
|
||||||
|
steps = workflow["jobs"]["smoke"]["steps"]
|
||||||
|
names = [s.get("name", "") for s in steps]
|
||||||
|
build_idx = None
|
||||||
|
scan_idx = None
|
||||||
|
for i, name in enumerate(names):
|
||||||
|
if "cmake" in name.lower() or "build" in name.lower():
|
||||||
|
if build_idx is None:
|
||||||
|
build_idx = i
|
||||||
|
if "secret" in name.lower():
|
||||||
|
scan_idx = i
|
||||||
|
if build_idx is not None and scan_idx is not None:
|
||||||
|
assert build_idx < scan_idx, (
|
||||||
|
"Build step should run before secret scan to fail fast on broken code"
|
||||||
|
)
|
||||||
Reference in New Issue
Block a user