cuda测试二维block的使用

时间:2023-03-09 09:39:17
cuda测试二维block的使用
#include "cuda_runtime.h"
#include <stdio.h>
#include <stdlib.h>
#include <math.h>
#include <memory.h> __global__ static void kernel(int *d_int){
int i;
int x = threadIdx.y;
for(i=;i<;i++){
d_int[i* + x] = i;
}
} int main(){
int gridsize = ;
dim3 blocksize(,);
int *h_int,*d_int;
h_int = (int *)malloc(sizeof(int)**);
cudaMalloc((void **) &d_int,sizeof(int)**);
cudaMemset(d_int,,sizeof(int)**);
memset(h_int,,sizeof(int)**);
kernel<<<gridsize,blocksize>>>(d_int);
cudaMemcpy(h_int,d_int,sizeof(int)**,cudaMemcpyDeviceToHost); for(int i=;i<;i++){
for(int j=;j<;j++){
printf("%d ",h_int[i*+j]);
}
printf("\n");
}
printf("\n");
getchar();
return ;
}

执行结果:

cuda测试二维block的使用