Blame for code/bugfind/bench/kernels/tract_len/post/src/lib.rs

pub fn tensor_len_ok(dims: &[usize], bits_per_item: usize, data_size_bytes: usize) -> bool {
    let mut len = 1usize;
    let mut overflow = false;
    let mut i = 0usize;
    while i < dims.len() {
        let d = dims[i];
        if d != 0 && len > usize::MAX / d {
            overflow = true;
        } else {
            len = len * d;
        }
        i += 1;
    }
    if overflow {
        return false;
    }
    let per = bits_per_item / 8;
    if per != 0 && len > usize::MAX / per {
        return false;
    }
    len * per == data_size_bytes
}