如何在CUDA的塊中返回其他線程？

Question

我正在嘗試訪問同一塊中的其他線程，但我想返回一些線程。 我想做這樣的事情：

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