Metalkit:MTLBuffer和swift 3中的指针

问题描述 投票:2回答:1

我从Metalkit开始,我有一个非常简单的内核作为测试用例。

kernel void compute(device float* outData [[ buffer(0) ]])
{
  outData[0] = 234.5;
  outData[3] = 345.6;
}

这个“计算”数据存储在MTLBuffer中。

var buffer : MTLBuffer?
...
buffer = device.makeBuffer(length: MemoryLayout<Float>.size * 5, options: [])
...
commandBuffer.waitUntilCompleted()

此时内核已经将一些测试数据写入MTLBuffer。问题是我应该如何从主程序访问该数据?

我从unsafeMutableRawPointer得到一个buffer.contents()。如何获得可以在其他地方使用的快速数组值(在屏幕上显示,写入文件,......)?

这些片段在这个非常简单的应用程序中工作,但我不确定它们是否正确:

let raw = buffer.contents()
let b = raw.bindMemory(to: Float.self, capacity: 5) 
print(b.advanced(by: 3).pointee)

let a = raw.assumingMemoryBound(to: Float.self)
print(a.advanced(by: 3).pointee)


let bufferPointer = UnsafeBufferPointer(start: b, count: 5)
let values = Array(bufferPointer)
print(values)


let value = raw.load(fromByteOffset: MemoryLayout<Float>.size * 3, as: Float.self)
print(value)

bindMemoryassumingMemoryBound都工作。虽然assumingMemoryBound假设底层字节已经打字,bindMemory没有。我认为其中一个应该有效,但不是两个都有效。应该是哪一个,为什么?

pointers memory-management swift3 metalkit
1个回答
1
投票

我使用下面提供的代码加载到数组,但我不能决定我或你的版本是否最好。

let count = 16

var array = [Float]()
array.reserveCapacity(count)

    for i in 0..<count {
        array.append(buffer.contents().load(fromByteOffset: MemoryLayout<Float>.size * i, as: Float.self))
    }
© www.soinside.com 2019 - 2024. All rights reserved.