cuda_kernel_call

Материал из CAMaaS preliminary wiki
Перейти к навигации Перейти к поиску
template <class F, class...Args>
void cuda_kernel_call(F func, unsigned cBlocks, unsigned cThreadsPerBlock, std::size_t cbSharedMemory, cudaStream_t stream, Args&&...args);

Если необходимо, создает и инициализирует и инициализирует разделяемое состояние для ядер CUDA, после чего производит вызов указанного ядра.

Параметры шаблона
FТип С функции, реализующей ядро со спецификатором __global__, которому делегируется вызов.
ArgsТипы аргументов args, которые передаются вызываемому ядру CUDA.
Параметры
funcС функция, заданная со спецификатором __global__, которой делегируется вызов.
cBlocksЧисло блоков CUDA, которые реализуют выполнение ядра func.
cThreadsPerBlockЧисло графических потоков в одном блоке CUDA, выполняющих ядро func.
cbSharedMemoryОбъем используемой ядром func разделяемой памяти.
streamПоток, используемый для вызова ядра func.
argsНабор аргументов, которые передаются ядру func при его вызове. Типы всех аргументов должны удовлетворять требованиям StandardLayoutType, иначе корректное поведение функции cuda_kernel_call и ядра не гарантируется. При вызове производится их автоматическое копирование в глобальную память текущего устройства CUDA.
См. также
cuda_stream_synchronizeСинхронизирует выполнение задач, связанных с указанным потоком CUDA stream, и производит дополнительную общую постобработку результатов.
cuda_abort_with_errorУстанавливает код ошибки с опциональным связанным текстовым описанием в глобальной памяти устройства, на котором производится вызов, и досрочно завершает выполнения ядра (kernel) CUDA.

Вызов

cuda_kernel_call(my_kernel, 10, 100, 0, cuda_stream().get(), x, y, z);

аналогичен вызову

my_kernel<<<10, 100, 0, cuda_stream.get()>>>(x, y, z);

с дополнительной инициализацией состояния, используемого реализацией CUDA.