c++ - 如何在 CUDA 的 block 中返回其他线程？

Question

我正在尝试访问同一个 block 中的其他线程，我想返回一些。我想做这样的事情:

#include "cuda_runtime.h"
#include "device_launch_parameters.h"
#include <stdio.h>

__global__ void returnKernel()
{
    if (blockIdx.x == threadIdx.x)
    {
        //exit/return thread 1
    }
}

int main()
{
    returnKernel<<<4, 4>>>();
    return 0;
}

提前致谢!

Answer 1

没有办法“杀死”其他正在运行的线程，它们需要“投降”自己。这是您的问题中的代码，已修改为从线程 1 退出:

#include "cuda_runtime.h"
#include "device_launch_parameters.h"
#include <stdio.h>

__global__ void returnKernel()
{
    if (blockIdx.x < blockDim.x && threadIdx.x == 1)
    {
        return;
    }
}

int main()
{
    returnKernel<<<4, 4>>>();
    return 0;
}