Filling an array concurrently

前端 未结 1 1972
我在风中等你
我在风中等你 2021-01-14 07:37

I am stumbling upon an issue with concurrency and arrays in Swift 5. To reproduce the issue I have simplified my code to the following fragment:

import Dispa         


        
相关标签:
1条回答
  • 2021-01-14 08:08

    There are pointers inside an Array that absolutely can change under your feet. It is not raw memory.

    Arrays are not thread-safe. Arrays are value types, which means that they support copy-on-write in a thread-safe way (so you can freely pass an array to another thread, and if it is copied there, that is ok), but you can't mutate the same array on multiple threads. An Array is not a C buffer. It's not promised to have contiguous memory. It's not even promised to allocate memory at all. Array could choose internally to store "I'm currently all zeros" as a special state and just return 0 for every subscript. (It doesn't, but it's allowed to.)

    For this specific problem, you'd typically use vDSP methods like vDSP_vramp, but I understand this is just an example, and there may not be a vDSP method that solves the problem. Typically, though, I'd still focus on Accelerate/SIMD methods rather than dispatching to queues.

    But if you are going to dispatch to queues, you'll need an UnsafeMutableBuffer to take control of the memory (and ensure that the memory even exists):

    pixels.withUnsafeMutableBufferPointer { pixelsPtr in
        DispatchQueue.concurrentPerform(iterations: threadCount) { thread in
            for number in thread*size ..< (thread+1)*size {
                let floating = Float(number)
                pixelsPtr[number] = SIMD3(floating, floating, floating)
            }
        }
    }
    

    The "Unsafe" indicates that it's now your problem to make sure all the accesses are legal and that you're not creating race conditions.

    Note the use of .concurrentPerform here. As @user3441734 reminds us, pixelsPtr is not promised to be valid once .withUnsafeMutablePointer completes. .concurrentPerform is guaranteed not to return until all blocks are complete, so the pointer is guaranteed to be valid.

    This could be done with DispatchGroup as well, but the .wait would need to be inside of withUnsafeMutableBufferPointer.

    0 讨论(0)
提交回复
热议问题