Usando lambdas, objetos de função e funções restritas
O código de C++ AMP que você deseja executar no acelerador é especificado como um argumento na chamada ao método parallel_for_each. Você pode fornecer uma expressão lambda ou um objeto de função (functor) como o argumento. Além de isso, a expressão lambda ou objeto de função podem chamar a função restrita C++ AMP. Este tópico usa um algoritmo de adição de matriz para demonstrar lambdas, objetos de função e funções restritas. O exemplo a seguir mostra o algoritmo sem código de C++ AMP. São criadas duas matrizes unidimensionais de tamanho igual. Os elementos inteiros correspondentes são adicionados e armazenados em uma terceira matriz unidimensional. O C++ AMP não é usado.
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";
}
}
Expressão lambda
Usar uma expressão lambda é a forma mais direta de usar o C++ AMP para reescrever o código.
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";
}
}
A expressão lambda deve incluir um parâmetro indexação e deve incluir restrict(amp). No exemplo, o objeto array_view sum tem classificação 1. Portanto, o parâmetro para a instrução lambda é um objeto índice que possui a posição 1. Em tempo de execução, a expressão lambda é executada uma vez para cada elemento no objeto array_view. Para obter mais informações, consulte Sintaxe da expressão lambda.
Objeto de função
É possível incluir o código de aceleradores em um objeto de função.
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";
}
}
O objeto de função deve incluir um construtor e incluir uma sobrecarga do operador de chamada de função. O operador de chamada de função deve incluir um parâmetro indexação. Uma instância do objeto de função é passada como o segundo argumento para o método parallel_for_each. Nesse exemplo, três objetos array_view são passados para o construtor do objeto de função. O objeto sum array_view tem classificação 1. Portanto, o parâmetro para o operador de chamada de função é um objeto índice que possui a posição 1. Em tempo de execução, a função é executada uma vez para cada elemento no objeto array_view. Para obter mais informações, consulte Chamada de função (C++) e Objetos de Função.
Função restrita C++ AMP
Você ainda pode incluir o código de aceleradores criando uma função restrita e chamando a partir de uma expressão lambda ou de um objeto de função. O exemplo de código a seguir demonstra como chamar uma função estritamente de uma expressão 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)
{
AddElementsWithRestrictedFunction(idx, sum, a, b);
}
);
for (int i = 0; i < 5; i++) {
std::cout << sum[i] << "\n";
}
}
A função estritamente deve incluir restrict(amp) e estar de acordo com as restrições que são descritas em Cláusula de restrição (AMP C++).
Consulte também
Referência
Cláusula de restrição (AMP C++)