debug for laion100m

tensorchord · Nov 8, 2024 · 59e5fa9 · 59e5fa9
1 parent d17f040
commit 59e5fa9
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git a/src/algorithm/insert.rs b/src/algorithm/insert.rs
@@ -121,6 +121,7 @@ pub fn insert(relation: Relation, payload: Pointer, vector: Vec<f32>, distance_k
                             &h2_tuple.factor_err,
                             &h2_tuple.t,
                         ),
+                        1.9,
                     );
                     for j in 0..32 {
                         if h2_tuple.mask[j] {
@@ -191,6 +192,7 @@ pub fn insert(relation: Relation, payload: Pointer, vector: Vec<f32>, distance_k
                             &h1_tuple.factor_err,
                             &h1_tuple.t,
                         ),
+                        1.9,
                     );
                     for j in 0..32 {
                         if h1_tuple.mask[j] {

diff --git a/src/index/am_options.rs b/src/index/am_options.rs
@@ -88,8 +88,8 @@ pub fn convert_opfamily_to_vd(
 fn convert_name_to_vd(name: &str) -> Option<(VectorKind, PgDistanceKind)> {
     match name.strip_suffix("_ops") {
         Some("vector_l2") => Some((VectorKind::Vecf32, PgDistanceKind::L2)),
-        Some("vector_ip") => Some((VectorKind::Vecf32, PgDistanceKind::Dot)),
-        Some("vector_cosine") => Some((VectorKind::Vecf32, PgDistanceKind::Cos)),
+        Some("vector_ip" | "vector_dot") => Some((VectorKind::Vecf32, PgDistanceKind::Dot)),
+        Some("vector_cosine" | "vector_cos") => Some((VectorKind::Vecf32, PgDistanceKind::Cos)),
         _ => None,
     }
 }

diff --git a/src/lib.rs b/src/lib.rs
@@ -19,6 +19,9 @@ pgrx::extension_sql_file!("./sql/finalize.sql", finalize);
 
 #[pgrx::pg_guard]
 unsafe extern "C" fn _PG_init() {
+    unsafe {
+        std::env::set_var("RUST_BACKTRACE", "1");
+    }
     if unsafe { pgrx::pg_sys::IsUnderPostmaster } {
         pgrx::error!("rabbithole must be loaded via shared_preload_libraries.");
     }