libcorenn 0.3.1

Library for CoreNN
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41

use super::Compressor;
use super::CV;
use crate::metric::StdMetric;
use crate::vec::VecData;
use std::sync::Arc;

#[derive(Debug)]
pub struct TruncCompressor {
  dim: usize,
}

impl TruncCompressor {
  pub fn new(dim: usize) -> Self {
    Self { dim }
  }
}

impl Compressor for TruncCompressor {
  fn into_compressed(&self, v: VecData) -> CV {
    macro_rules! trunc {
      ($a:expr) => {{
        $a.truncate(self.dim);
        $a
      }};
    }
    let v = match v {
      VecData::BF16(mut a) => VecData::BF16(trunc!(a)),
      VecData::F16(mut a) => VecData::F16(trunc!(a)),
      VecData::F32(mut a) => VecData::F32(trunc!(a)),
      VecData::F64(mut a) => VecData::F64(trunc!(a)),
    };
    Arc::new(v)
  }

  fn dist(&self, metric: StdMetric, a: &CV, b: &CV) -> f64 {
    let a = a.downcast_ref::<VecData>().unwrap();
    let b = b.downcast_ref::<VecData>().unwrap();
    let f = metric.get_fn();
    f(a, b)
  }
}