ios - 在Swift中从MTLBuffer读取数据

在GPU上进行处理后，我需要从MTLBuffer中读取数据。到目前为止，我已经尝试了以下代码，但始终会因EXC_BAD_ACCESS错误代码而崩溃。

struct gauss_model {
  var mean : [Float32] = [0.0, 0.0, 0.0];
  var covMat : [[Float32]] = [[0.0, 0.0, 0.0],[0.0, 0.0, 0.0],[0.0, 0.0, 0.0]];
  var invCovMat : [[Float32]] = [[0.0, 0.0, 0.0],[0.0, 0.0, 0.0],[0.0, 0.0, 0.0]];
  var samples : Int32 = 0;
}

self.gaussModels = [gauss_model](count: Int(10), repeatedValue: gauss_model())
self.modelsBuffer = self.device.newBufferWithBytes(self.gaussModels, length:      self.gaussModels.count * sizeof(gauss_model), options: MTLResourceOptions.OptionCPUCacheModeDefault)
commandEncoder.setBuffer(self.modelsBuffer, offset: 0, atIndex: 0)

    // execute GPU code

var model = unsafeBitCast(self.modelsBuffer.contents(), UnsafeMutablePointer<gauss_model>.self)
NSLog("%@", model.memory.mean) // crashes on this statement

我还尝试了获得价值的不同方法，例如

var model = UnsafeMutablePointer<gauss_model>(self.modelsBuffer.contents())
// iterate over models with model.memory and model.successor()

要么

var model = UnsafeMutablePointer<[gauss_model]>(self.modelBuffer.contents())
let models : [gauss_model] = model.memory

但他们都不起作用。有没有办法做到这一点？

最佳答案

我设法解决了这个问题。问题出在我对Swift中的内存管理的错误假设以及newBufferWithBytes函数仅执行浅表复制这一事实。调用该函数只会将指针复制到mean，covMat和invCovMat数组，而self.modelBuffer.contents()包含指向未初始化内存的指针。访问内存导致崩溃。

关于ios - 在Swift中从MTLBuffer读取数据，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/27708228/