CUDA笔记2

最新推荐文章于 2024-11-13 10:58:56 发布

捣蛋老爷

最新推荐文章于 2024-11-13 10:58:56 发布

阅读量276

点赞数

分类专栏： # 模型部署文章标签：笔记

本文链接：https://blog.csdn.net/hukou6335490/article/details/130182588

版权

模型部署专栏收录该内容

4 篇文章 0 订阅

订阅专栏

1.硬件理解

1.1对应

在这里插入图片描述

1.2 不一定是同时执行

例如只有13个sm,每个sm有128个core,而我们创建了1百万个threads,就要同步执行
因此，我们倾向于在block的x维设置为32的倍数,防止浪费warp
warp id打印

#include <stdio.h>
#include <stdlib.h>

#include "cuda_runtime.h"
#include "device_launch_parameters.h"

__global__ void print_details_of_warps()
{
	int gid = blockIdx.y * gridDim.x * blockDim.x 
		+ blockIdx.x * blockDim.x + threadIdx.x;

	int warp_id = threadIdx.x / 32;

	int gbid = blockIdx.y * gridDim.x + blockIdx.x;

	printf("tid : %d, bid.x : %d, bid.y : %d, gid : %d, warp_id : %d, gbid : %d \n",
		threadIdx.x, blockIdx.x, blockIdx.y, gid, warp_id, gbid);
}

int main(int argc , char** argv)
{
	dim3 block_size(42);
	dim3 grid_size(2,2);

	print_details_of_warps << <grid_size,block_size >> > ();
	cudaDeviceSynchronize();

	cudaDeviceReset();
	return EXIT_SUCCESS;
}