library CB-Console;
library CB-GPU;
library CB-GenerationOptimized;

fn main() {
    string dataset = "data/ai/shared-10gb-dataset.txt";
    string index = "data/ai/shared-10gb-dataset.cbi";
    string tags = "ai,index,generation";

    CB-GPU.set_budget_mb(4096);
    u256 started = time.nowMs();
    array records = CB-GenerationOptimized.query_gpu(dataset, index, tags, 8);
    u256 elapsed = time.nowMs() - started;
    if (elapsed == 0) { elapsed = 1; }

    println("ClosedBit Shared 10 GiB Direct3D 12 Retrieval");
    println("Adapter: " + CB-GPU.name());
    println("Backend: " + CB-GenerationOptimized.last_backend());
    println("VRAM policy cap MB: " + CB-GPU.budget_mb());
    println("Indexed records: " + CB-GenerationOptimized.count(index));
    println("Matches returned: " + records.length());
    println("AI_GPU_BENCH|closedbit|" + CB-GenerationOptimized.count(index) + "|" + records.length() + "|" + elapsed + "|" + CB-GenerationOptimized.last_backend());
}