#include <stdio.h>
__global__ void childKernel(int i)
{
int tid = blockIdx.x*blockDim.x+threadIdx.x;
printf("parent:%d,child:%d\n",i,tid);
for(int j=i;j<i+10;j++)
{
printf(",%d",j);
}
printf("\n");
}
__global__ void kernel()
{
int tid = blockIdx.x*blockDim.x+threadIdx.x;
childKernel<<<1,2>>>(tid);
}
int main()
{
kernel<<<1,1>>>();
cudaDeviceSynchronize();
return 0;
}
cuda核函数再调用核函数,多层并行
最新推荐文章于 2025-05-14 22:53:34 发布
本文介绍如何使用CUDA在GPU上实现并行计算,通过`__global__`函数和`kernel`函数进行GPU编程,展示如何在父线程中调用子线程函数,并在GPU上执行大量数据处理任务。
部署运行你感兴趣的模型镜像
您可能感兴趣的与本文相关的镜像
PyTorch 2.7
PyTorch
Cuda
PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理
1127





