Function main

org/src/main.rs:63–111

()

}

fn main() {
    // Initialize the lookup table once
    initialize_u16_lookup_table();
    println!("done");

    let size = 64;
    let min = 0.0;
    let max = 1.0;
    let vec1 = (0..size)
        .map(|_| {
            let mut rng = rand::thread_rng();
            let random_number: f32 = rng.gen_range(min..max);
            random_number
        })
        .collect::<Vec<f32>>();

    let vec2 = (0..size)
        .map(|_| {
            let mut rng = rand::thread_rng();
            let random_number: f32 = rng.gen_range(min..max);
            random_number
        })
        .collect::<Vec<f32>>();

    println!("raw vec A :{:?}", vec1);
    println!("raw vec B :{:?}", vec2);

    let resolution = 1 as u8;
    let quantized_values1: Vec<Vec<u32>> = quantize_to_u32_bits(&vec1.clone(), resolution);
    let quantized_values2: Vec<Vec<u32>> = quantize_to_u32_bits(&vec2.clone(), resolution);

    let vector_list1 = VectorQt {
        quant_vec: quantized_values1,
        resolution: resolution,
    };
    let vector_list2 = VectorQt {
        quant_vec: quantized_values2,
        resolution: resolution,
    };
    println!("quantized vec A :{:?}", vector_list1);
    println!("quantized vec B :{:?}", vector_list2);

    let normal = cosine_similarity(&vec1, &vec2);
    println!("cs <normal>: {}", normal);
    println!("\n\n");

    let scalar_quant_cs = cosine_similarity_new(&vector_list1, &vector_list2);
    println!("scalar_quant_cs : {}", scalar_quant_cs);
}

fn precompute_lookup_table() -> [[u8; 16]; 16] {
    let mut table = [[0; 16]; 16];