從我的Metal內核函數調用外部函數

Question

我有一個簡單的Metal文件，並且在函數內部，我想從要導入的C ++庫（ interface.h ）中調用一個耗時的函數。

我的目標是將signBlock函數內部發生的繁重工作從CPU轉移到GPU。

我所描述的可能嗎？ 這是一些代碼（公認不是很有用）

#include <metal_stdlib>

#include "interface.h"

using namespace metal;

kernel void testFunc();
{
   // In here, can I call function `signBlock` from `interface.h`?
}

Answer 1

如果signBlock()的代碼與Metal兼容，則可以調用它。 但是金屬着色語言與C ++並非100％相同，因此它不會僅接受任何C ++函數。 （並且代碼實際上需要存在於.metal源文件中，而不是存在於某些C ++庫中。）