在金属着色器代码中，如何定义函数的输入/输出参数变量？

Question

我有这个功能：

void myFunct(const int A, ?out? int B){B = B + A;} 

如何以函数可以更新其值并将其返回给调用者的方式声明 B？

Answer 1

Metal 是一种基于 C++ 的编程语言，它允许您将指针传递给函数。 为此，只需将函数参数声明为指针类型。

void myFunct(const int A, device int *B)
{
   *B = *B + A;
} 

device int *C = 0;
myFunct(1, C);

您还可以通过引用传递变量：

void myFunct(const int A, device int &B)
{
   B = B + A;
} 

device int *C = 0;
myFunct(1, *C)

或通过引用指针：

void myFunct(const int A, device int *&B)
{
   *B = *B + A;
} 

device int *C = 0;
myFunct(1, C);