gpucoder.atomicMax

Атомарно найдите максимум между заданным значением и переменной в глобальной памяти или общей памяти

свернуть все на странице

    Синтаксис

    [A,oldA] = gpucoder.atomicMax(A,B)

    Описание

    пример

    [A,oldA] = gpucoder.atomicMax(A,B) сравнивает B к значению A в глобальной памяти или общей памяти и записях max(A,B) назад в A. Операция является атомарной в некотором смысле, что целая операция "чтение изменяет запись", как, гарантируют, будет выполнен без интерференции от других потоков. Порядок аргументов ввода и вывода должен совпадать с обеспеченным синтаксисом.

    Примеры

    свернуть все

    Выполните простую атомарную операцию сложения при помощи gpucoder.atomicMax функционируйте и сгенерируйте CUDA® код, который вызывает соответствующий atomicMax() CUDA API.

    В одном файле запишите функции точки входа myAtomicMax это признает, что матрица вводит a и b.

    function a = myAtomicMax(a,b)

coder.gpu.kernelfun;
for i =1:numel(a)
    [a(i),~] = gpucoder.atomicMax(a(i), b);
end

end

    Создать тип для матрицы удваивается для использования в генерации кода, используйте coder.newtype функция.

    A = coder.newtype('int32', [1 30], [0 1]);
B = coder.newtype('int32', [1 1], [0 0]);
inputArgs = {A,B};

    Чтобы сгенерировать библиотеку CUDA, используйте codegen функция.

    cfg = coder.gpuConfig('lib');
cfg.GenerateReport = true;

codegen -config cfg -args inputArgs myAtomicMax -d myAtomicMax

    Сгенерированный код CUDA содержит myAtomicMax_kernel1 ядро с вызовами atomicMax() API CUDA.

    //
// File: myAtomicMax.cu
//
...

static __global__ __launch_bounds__(1024, 1) void myAtomicMax_kernel1(
    const int32_T b, const int32_T i, int32_T a_data[])
{
  uint64_T loopEnd;
  uint64_T threadId;
...

  for (uint64_T idx{threadId}; idx <= loopEnd; idx += threadStride) {
    int32_T b_i;
    b_i = static_cast<int32_T>(idx);
    atomicMax(&a_data[b_i], b);
  }
}
...

void myAtomicMax(int32_T a_data[], int32_T a_size[2], int32_T b)
{
  dim3 block;
  dim3 grid;
...

    cudaMemcpy(gpu_a_data, a_data, a_size[1] * sizeof(int32_T),
               cudaMemcpyHostToDevice);
    myAtomicMax_kernel1<<<grid, block>>>(b, i, gpu_a_data);
    cudaMemcpy(a_data, gpu_a_data, a_size[1] * sizeof(int32_T),
               cudaMemcpyDeviceToHost);
...

}

    Входные параметры

    свернуть все

    Операнды в виде скаляров, векторов, матриц или многомерных массивов. Входные параметры A и B должен удовлетворить следующим требованиям:

    • Имейте совпадающий тип данных.

    • Имейте тот же размер или имейте размеры, которые совместимы. Например, A M- N матрица и B скаляр или 1- N вектор-строка.

    Типы данных: int32 | uint32 | uint64

    Ограничения

    • Вход указателя на функцию к gpucoder.stencilKernel прагма не может содержать вызовы атомарных функций. Например,

      out1 = gpucoder.stencilKernel(@myAtomicMax,A,[3 3],'same',B);

    Смотрите также

    Функции

    Темы

    Введенный в R2021b

    Документация GPU Coder

    Поддержка

    Для просмотра документации необходимо авторизоваться на сайте