使用 Lambda、函式物件和限制函式
您想要在加速器上執行的 AMP C++ 程式碼會被指定為引數傳遞至 parallel_for_each 方法的呼叫。 您可以提供 Lambda 運算式或函式物件 (functor) 做為該引數。 此外，lambda 運算式或函式物件可以呼叫受限的函式。 本主題會使用陣列加法演算法，來示範 lambda、 函式物件，以及受限制的函式。 下列範例會示範不具有 C + + AMP 程式碼的演算法。 已建立長度相同的兩個一維陣列。 加入和儲存於第三個一維陣列中對應的整數項目。 不會使用 C + + AMP。
void CpuMethod() {
int aCPP[] = {1, 2, 3, 4, 5};
int bCPP[] = {6, 7, 8, 9, 10};
int sumCPP[5];
for (int idx = 0; idx < 5; idx++)
{
sumCPP[idx] = aCPP[idx] + bCPP[idx];
}
for (int idx = 0; idx < 5; idx++)
{
std::cout << sumCPP[idx] << "\n";
}
}
Lambda 運算式
使用 lambda 運算式是使用 C + + AMP 來重寫程式碼最直接的方式。
void AddArraysWithLambda() {
int aCPP[] = {1, 2, 3, 4, 5};
int bCPP[] = {6, 7, 8, 9, 10};
int sumCPP[5];
array_view<const int, 1> a(5, aCPP);
array_view<const int, 1> b(5, bCPP);
array_view<int, 1> sum(5, sumCPP);
sum.discard_data();
parallel_for_each(
sum.extent,
[=](index<1> idx) restrict(amp)
{
sum[idx] = a[idx] + b[idx];
}
);
for (int i = 0; i < 5; i++) {
std::cout << sum[i] << "\n";
}
}
Lambda 運算式必須包含一個索引參數，而且必須包含restrict(amp)。 在範例中，array_viewsum 物件排名為 1。 因此， lambda 陳述式的參數是一個索引物件，其排序為 1 。 在執行階段，lambda 運算式會為每個項目在array_view 物件中，執行一次。 如需詳細資訊，請參閱Lambda 運算式語法。
Function 物件
插入函式物件，就可以把加速器對應到程式碼。
class AdditionFunctionObject
{
public:
AdditionFunctionObject(const array_view<int, 1>& a,
const array_view<int, 1>& b,
const array_view<int, 1>& sum
)
: a(a), b(b), sum(sum)
{
}
void operator()(index<1> idx) restrict(amp)
{
sum[idx] = a[idx] + b[idx];
}
private:
array_view<int, 1> a;
array_view<int, 1> b;
array_view<int, 1> sum;
};
void AddArraysWithFunctionObject() {
int aCPP[] = {1, 2, 3, 4, 5};
int bCPP[] = {6, 7, 8, 9, 10};
int sumCPP[5];
array_view<const int, 1> a(5, aCPP);
array_view<const int, 1> b(5, bCPP);
array_view<int, 1> sum(5, sumCPP);
sum.discard_data();
parallel_for_each(
sum.extent,
AdditionFunctionObject(a, b, sum)
);
for (int i = 0; i < 5; i++) {
std::cout << sum[i] << "\n";
}
}
函式物件必須包含一個建構函式，而且必須包含函式呼叫運算子多載。 函式呼叫運算子必須包含一個索引參數。 函式物件的執行個體被傳遞做為第二個引數給 parallel_for_each 方法。 在這個範例中，三個 array_view 物件會傳遞至函式物件的建構函式。 array_viewsum 物件排名為 1。 因此，函式呼叫運算子的參數是索引排序 1 的物件。 在執行階段，此函式會對在 array_view 物件中的每個項目執行一次。 如需詳細資訊，請參閱函式呼叫 (C++)與函式物件。
C++ AMP 有限的函式
您可以進一步把加速器對應程式碼，藉由建立受限的功能，並從 lambda 運算式或函式物件呼叫。 下列程式碼範例示範如何從 lambda 運算式呼叫受限的功能。
void AddElementsWithRestrictedFunction(index<1> idx, array_view<int, 1> sum, array_view<int, 1> a, array_view<int, 1> b) restrict(amp)
{
sum[idx] = a[idx] + b[idx];
}
void AddArraysWithFunction() {
int aCPP[] = {1, 2, 3, 4, 5};
int bCPP[] = {6, 7, 8, 9, 10};
int sumCPP[5];
array_view<int, 1> a(5, aCPP);
array_view<int, 1> b(5, bCPP);
array_view<int, 1> sum(5, sumCPP);
sum.discard_data();
parallel_for_each(
sum.extent,
[=](index<1> idx) restrict(amp)
{
AddElements(idx, sum, a, b);
}
);
for (int i = 0; i < 5; i++) {
std::cout << sum[i] << "\n";
}
}
受限制的函式必須包含restrict(amp)且符合所述的限制限制子句 (C++ AMP)。