Question

我需要在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

但它们都没有奏效。有没有办法做到这一点？

Answer 1

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

从Swift中的MTLBuffer读取数据

1 个答案: