add hunyuan_ocr

tests/messy_test.rs

@@ -1,4 +1,6 @@
-use aha::utils::tensor_utils::interpolate_bicubic;
+use std::time::Instant;
+
+use aha::utils::tensor_utils::interpolate_bilinear;
 use anyhow::Result;
 use candle_core::Tensor;
 
@@ -6,11 +8,24 @@ use candle_core::Tensor;
 fn messy_test() -> Result<()> {
     // RUST_BACKTRACE=1 cargo test -F cuda messy_test -r -- --nocapture
     let device = &candle_core::Device::Cpu;
-    // let t = Tensor::randn(0.0f32, 1.0, (1, 768, 64, 64), device)?;
-    let t = Tensor::arange(0.0f32, 10.0, device)?.broadcast_as((1, 1, 10, 10))?;
+    let t = Tensor::arange(0.0f32, 40.0, device)?.broadcast_as((1, 1, 40, 40))?;
     println!("t: {}", t);
-    let t_resized = interpolate_bicubic(&t, (5, 5), Some(true), Some(false))?;
-    println!("t_resized: {}", t_resized);
+    let i_start = Instant::now();
+    let t_inter = interpolate_bilinear(&t, (20, 20), Some(false))?;
+    let i_duration = i_start.elapsed();
+    println!("Time elapsed in interpolate_bilinear is: {:?}", i_duration);
+    println!("t_inter: {}", t_inter);
+    // let x: Vec<u32> = (0..5).flat_map(|_| 0u32..10).collect();
+    // let id: Vec<u32> = (0..5).flat_map(|h| vec![h; 10]).collect();
+    // println!("x: {:?}", id);
+    // let t = Tensor::randn(0.0f32, 1.0, (1, 768, 64, 64), device)?;
+    // let t = Tensor::arange(0u32, 10, device)?.broadcast_as((1, 10))?;
+    // let eq = t.broadcast_eq(&Tensor::new(5u32, device)?)?;
+    // println!("eq: {}", eq);
+    // let t = Tensor::arange(0.0f32, 10.0, device)?.broadcast_as((1, 1, 10, 10))?;
+    // println!("t: {}", t);
+    // let t_resized = interpolate_bicubic(&t, (5, 5), Some(true), Some(false))?;
+    // println!("t_resized: {}", t_resized);
     // let t1 = Tensor::rand(0.0, 1.0, (1, 5, 5, 10), device)?;
     // let t2 = Tensor::rand(0.0, 1.0, (5, 8, 10), device)?;
     // let t2 = t2.t()?;