Function assert_close

cake-core/tests/unit_tests/test_backends.rs:164–175

(cpu: &Tensor, metal: &Tensor, tol: f32, label: &str)

#[cfg(feature = "metal")]
fn assert_close(cpu: &Tensor, metal: &Tensor, tol: f32, label: &str) {
    let cpu_vals: Vec<f32> = cpu.to_dtype(DType::F32).unwrap().flatten_all().unwrap().to_vec1().unwrap();
    let metal_vals: Vec<f32> = metal.to_device(&Device::Cpu).unwrap().to_dtype(DType::F32).unwrap().flatten_all().unwrap().to_vec1().unwrap();
    assert_eq!(cpu_vals.len(), metal_vals.len(), "{label}: length mismatch");
    for (i, (c, m)) in cpu_vals.iter().zip(metal_vals.iter()).enumerate() {
        assert!(
            (c - m).abs() < tol,
            "{label}[{i}]: cpu={c} metal={m} diff={}",
            (c - m).abs()
        );
    }
}

#[cfg(feature = "metal")]
#[test]