gpucoder.atomicMin

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

    Описание

    пример

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

    Примеры

    свернуть все

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

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

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

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

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

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

    cfg = coder.gpuConfig('lib');
    cfg.GenerateReport = true;
    
    codegen -config cfg -args inputArgs myAtomicMin -d myAtomicMin
    

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

    //
    // File: myAtomicMin.cu
    //
    ...
    
    static __global__ __launch_bounds__(1024, 1) void myAtomicMin_kernel1(
        const uint32_T b, const int32_T i, uint32_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);
        atomicMin(&a_data[b_i], b);
      }
    }
    ...
    
    void myAtomicMin(uint32_T a_data[], int32_T a_size[2], uint32_T b)
    {
      dim3 block;
      dim3 grid;
    ...
    
        cudaMemcpy(gpu_a_data, a_data, a_size[1] * sizeof(uint32_T),
                   cudaMemcpyHostToDevice);
        myAtomicMin_kernel1<<<grid, block>>>(b, i, gpu_a_data);
        cudaMemcpy(a_data, gpu_a_data, a_size[1] * sizeof(uint32_T),
                   cudaMemcpyDeviceToHost);
    ...
    
    }
    

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

    свернуть все

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

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

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

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

    Ограничения

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

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

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