Function make_f16_tensor

cake-core/tests/protocol.rs:21–28

(shape: &[usize])

// ============================================================================

fn make_f16_tensor(shape: &[usize]) -> Tensor {
    let n: usize = shape.iter().product();
    let data: Vec<f32> = (0..n).map(|i| i as f32 * 0.01).collect();
    Tensor::from_vec(data, shape, &Device::Cpu)
        .unwrap()
        .to_dtype(DType::F16)
        .unwrap()
}

fn mock_worker_info(latency: u128) -> WorkerInfo {
    WorkerInfo {