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

2022-10-12 22:20:08 +08:00
 0Vincent0Zhang0
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 做了多次的写引用。

谢谢!

916 次点击
所在节点    Rust
1 条回复
0Vincent0Zhang0
2022-10-13 08:56:09 +08:00
已解决
'''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 在多线程时的生命周期问题。

这是一个专为移动设备优化的页面(即为了让你能够在 Google 搜索结果里秒开这个页面),如果你希望参与 V2EX 社区的讨论,你可以继续到 V2EX 上打开本讨论主题的完整版本。

https://www.v2ex.com/t/886482

V2EX 是创意工作者们的社区,是一个分享自己正在做的有趣事物、交流想法,可以遇见新朋友甚至新机会的地方。

V2EX is a community of developers, designers and creative people.

© 2021 V2EX