Непоследовательные результаты при использовании Metal Performance Shaders между MacBook Pro и iMac

Question

Я пытаюсь окунуться в воду программирования на GPU впервые. Я думал, что начну с чего-то простого и использую готовые ядра (отсюда и MPS) и просто попытаюсь выдать команды для GPU.

Моя попытка состояла в том, чтобы просто сложить все значения от 1 до 1000. Я поместил каждое значение в матрицу 1x1 и использовал матричную сумму MPS.

На моем MacBook Pro это работает так, как я ожидаю.

На моем iMac это дает [0.0] как результат. Я подумал, что это связано с памятью, так как я использую iGPU на своем MacBook Pro и dGPU на своем iMac, однако, насколько я могу судить, storageModeShared не должен приводить к этому. Я даже пытался добавить.synchronize() к матрице результатов, прежде чем пытаться читать из нее, хотя я почти уверен, что это не должно быть необходимо с storageModeShared.

Код не изящен, потому что он предназначен только для быстрого понимания работы команд с MPS, и я пытался исправить проблемы некоторое время, не отслеживая структуру, но его все равно должно быть довольно легко прочитать; Если нет, дайте мне знать, и я сделаю рефакторинг.

Операторы печати предназначены только для отладки, кроме печати (вывод)

Я ненавижу вставлять столько кода, но боюсь, что не смогу больше изолировать свою проблему.

import Cocoa
import Quartz
import PlaygroundSupport
import MetalPerformanceShaders

let device = MTLCopyAllDevices()[0]
print(MTLCopyAllDevices())


let shaderKernel = MPSMatrixSum.init(device: device, count: 1000, rows: 1, columns: 1, transpose: false)


var matrixList: [MPSMatrix] = []
var GPUStorageBuffers: [MTLBuffer] = []
for i in 1...1000 {
    var a = Float32(i)
    var b: [Float32] = []
    let descriptor = MPSMatrixDescriptor.init(rows: 1, columns: 1, rowBytes: 4, dataType: .float32)
    b.append(a)

    let buffer = device.makeBuffer(bytes: b, length: 4, options: .storageModeShared)

    GPUStorageBuffers.append(buffer!)

    let GPUStoredMatrices = MPSMatrix.init(buffer: buffer!, descriptor: descriptor)

    matrixList.append(GPUStoredMatrices)
}

let matrices: [MPSMatrix] = matrixList

print(matrices.count)
print("\n")

print(matrices[4].debugDescription)
print("\n")
var printer: [Float32] = []
let pointer2 = matrices[4].data.contents()
let typedPointer2 = pointer2.bindMemory(to: Float32.self, capacity: 1)
let buffpoint2 = UnsafeBufferPointer(start: typedPointer2, count: 1)

buffpoint2.map({value in
    printer += [value]
})
print(printer)


let CMDQue = device.makeCommandQueue()

let CMDBuffer = CMDQue!.makeCommandBuffer()

var resultMatrix = MPSMatrix.init(device: device, descriptor: MPSMatrixDescriptor.init(rows: 1, columns: 1, rowBytes: 4, dataType: .float32))

shaderKernel.encode(to: CMDBuffer!, sourceMatrices: matrices, resultMatrix: resultMatrix, scale: nil, offsetVector: nil, biasVector: nil, start: 0)

print(CMDBuffer.debugDescription)
CMDBuffer!.commit()
print(CMDBuffer.debugDescription)
print(CMDQue.debugDescription)
let GPUStartTime = CACurrentMediaTime()
CMDBuffer!.waitUntilCompleted()

var output = [Float32]()
resultMatrix.synchronize(on: CMDBuffer!)

let pointer = resultMatrix.data.contents()
let typedPointer = pointer.bindMemory(to: Float32.self, capacity: 1)
let buffpoint = UnsafeBufferPointer(start: typedPointer, count: 1)

buffpoint.map({value in
    output += [value]
})
print(output)
let finish = GPUStartTime - CACurrentMediaTime()
print("\n")
print(finish)