在金屬着色器代碼中，如何定義函數的輸入/輸出參數變量？

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);