V2EX = way to explore
V2EX 是一个关于分享和探索的地方
现在注册
已注册用户请  登录
0Vincent0Zhang0
V2EX  ›  Rust

请问怎样对一个数组按大小切出多个可写的引用分片?

  •  
  •   0Vincent0Zhang0 · 2022-10-12 22:20:08 +08:00 · 1143 次点击
    这是一个创建于 806 天前的主题,其中的信息可能已经有所发展或是发生改变。
    struct Slice<'a> {
        index: usize,
        data: &'a mut [i32]
    }
    
    fn block_slice(block: &mut [i32], slice_size: usize) -> Vec<Slice> {
        let slice_num =
            if block.len() % slice_size == 0 {
                block.len() / slice_size
            } else {
                block.len() / slice_size + 1
            };
        if slice_num == 1 {
            return vec![Slice { index: 0, data: block }];
        }
        let mut slices = Vec::with_capacity(slice_num);
        let mut rest = block;
        for index in 1..slice_num {
            let (head, tail) = rest.split_at_mut(index * slice_size);
            slices.push(Slice { index: index - 1, data: head });
            if index == slice_num - 1 {
                slices.push(Slice { index, data: tail });
            } else {
                rest = tail;
            }
        }
        return slices;
    }
    

    大佬们好!

    请问怎样可以对一个数组按大小切出多个可写的引用分片?(目的是准备把这些分片做并行修改)

    上面的代码有问题,第 20 行的 rest 做了多次的写引用。

    谢谢!

    0Vincent0Zhang0
        1
    0Vincent0Zhang0  
    OP
       2022-10-13 08:56:09 +08:00   ❤️ 1
    已解决
    '''rust
    let mut block = [0i32; BLOCK_SIZE];
    let chunks: Vec<&mut [i32]> = block.chunks_mut(SLICE_SIZE).collect();
    crossbeam::scope(|spwawner| {
    for (chunk_index, chunk) in chunks.into_iter().enumerate() {
    spwawner.spawn(move || {
    for index in 0..chunk.len() {
    chunk[index] += (index as i32).pow(2);
    }
    println!("Chunk[{:02}] completed!", chunk_index);
    });
    }
    });
    '''
    使用 chunks_mut 可以按大小分出可修改的切片,另外还需要 crossbeam 处理 block 在多线程时的生命周期问题。
    关于   ·   帮助文档   ·   博客   ·   API   ·   FAQ   ·   实用小工具   ·   1223 人在线   最高记录 6679   ·     Select Language
    创意工作者们的社区
    World is powered by solitude
    VERSION: 3.9.8.5 · 19ms · UTC 23:27 · PVG 07:27 · LAX 15:27 · JFK 18:27
    Developed with CodeLauncher
    ♥ Do have faith in what you're doing.