Have added github repo with corresponding Swift/Metal code - https://github.com/atveit/SwiftMetalGPUParallelProcessing