Używanie wyrażeń lambda, obiektów Function i funkcji z ograniczeniami

Kod C++ AMP, który chcesz uruchomić na akceleratorze, jest określony jako argument w wywołaniu metody parallel_for_each.Możesz dostarczyć wyrażenie lambda lub obiekt funkcyjny (funktor) w roli tego argumentu.Ponadto wyrażenie lambda lub obiekt funkcyjny mogą wywołać funkcję C++ AMP z ograniczeniami.Ten temat korzysta z algorytmu dodawania tablic do zademonstrowania wyrażeń lambda, obiektów funkcyjnych i funkcji z ograniczeniami.W poniższym przykładzie pokazany jest algorytm bez kodu C++ AMP.Tworzone są dwie tablice 1-wymiarowe równej długości.Odpowiadające elementy całkowite są dodawane i przechowywane w trzeciej 1-wymiarowej tablicy.C++ AMP nie jest używane.

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";
    }
}

Wyrażenie lambda

Użycie wyrażenia lambda jest najbardziej bezpośrednim sposobem wykorzystania C++ AMP do przepisania kodu.

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";
    }
}

Wyrażenie lambda musi zawierać jeden parametr indeksowania i musi zawierać restrict(amp).Na przykład obiekt array_viewsum ma rangę 1.Dlatego parametr do instrukcji lambda jest obiektem index o randze 1.W czasie wykonywania, wyrażenie lambda jest wykonywane raz dla każdego elementu w obiekcie array_view.Aby uzyskać więcej informacji, zobacz Składnia wyrażenia lambda.

Obiekt funkcji

Możesz przekazać kod akceleratora do obiektu funkcyjnego.

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";
    }
}

Obiekt funkcyjny musi zawierać konstruktor i przeciążenie operatora wywołania funkcji.Operator wywołania funkcji musi zawierać jeden parametr indeksowania.Wystąpienie obiektu funkcyjnego jest przekazywane jako drugi argument do metody parallel_for_each.W tym przykładzie trzy obiekty array_view są przekazywane do funkcji konstruktora obiektu.Obiekt array_viewsum ma rangę 1.Dlatego parametr operatora wywołania funkcji jest obiektem index o randze 1.W czasie wykonywania, funkcja jest wykonywana raz dla każdego elementu w obiekcie array_view.Aby uzyskać więcej informacji, zobacz Wywołanie funkcji (C++) i Obiekty funkcji.

Funkcja C++ AMP z ograniczeniami

Możesz przekazać kod akceleratora tworząc funkcję z ograniczeniami i wywołując ją w wyrażeniu lambda lub w obiekcie funkcyjnym.Poniższy przykład kodu pokazuje sposób wywołania funkcji z ograniczeniami z wyrażenia 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";
    }
}

Funkcja z ograniczeniami musi zawierać restrict(amp) i być zgodna z ograniczeniami, które są opisane w Klauzula ograniczenia (C++ AMP).

Zobacz też

Informacje

Składnia wyrażenia lambda

Wywołanie funkcji (C++)

Klauzula ograniczenia (C++ AMP)

Koncepcje

Obiekty funkcji

Inne zasoby

C++ AMP (C++ Accelerated Massive Parallelism)