zoukankan      html  css  js  c++  java
  • 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=0;i<3;i++){
            d_int[i*4 + x] = i;
        }
    }
    
    int main(){
        int gridsize = 1;
        dim3 blocksize(3,4);
        int *h_int,*d_int;
        h_int = (int *)malloc(sizeof(int)*3*4);
        cudaMalloc((void **) &d_int,sizeof(int)*3*4);
        cudaMemset(d_int,0,sizeof(int)*3*4);
        memset(h_int,0,sizeof(int)*3*4);
        kernel<<<gridsize,blocksize>>>(d_int);
        cudaMemcpy(h_int,d_int,sizeof(int)*3*4,cudaMemcpyDeviceToHost);
    
        for(int i=0;i<3;i++){
            for(int j=0;j<4;j++){
                printf("%d ",h_int[i*4+j]);
            }
            printf("
    ");
        }
        printf("
    ");
        getchar();
        return 0;
    }

    执行结果:

  • 相关阅读:
    php 解析xml
    php
    php 设置自动加载某个页面
    Mac
    mysql
    Git
    C#
    C# 正则表达式
    C# ASCII码排序
    (转)datagridview 自定义列三步走
  • 原文地址:https://www.cnblogs.com/xing901022/p/3464798.html
Copyright © 2011-2022 走看看