update check_dimension

Author: hisashi-ito

README.md

@@ -12,4 +12,6 @@
 * hello  
 hello world コード
 * sumArray  
-ベクトルの和のサンプル
+ベクトルの和のサンプル  
+* check_dimension
+グリッド,ブロック,スレッドの確認サンプル

check_dimension/Makefile

@@ -0,0 +1,3 @@
+# Make file for check_dimension.cu
+check_dimension:check_dimension.cu
+	nvcc -Wno-deprecated-gpu-targets -o check_dimension check_dimension.cu

check_dimension/check_dimension

@@ -0,0 +1,41 @@
+#include <cuda_runtime.h>
+#include <stdio.h>
+
+__global__ void checkIndex(void){
+  printf("threadIdx: (%d,%d,%d) blockIdx: (%d, %d, %d) blockDim: (%d, %d, %d) gridDim: (%d, %d, %d)\n",
+	 threadIdx.x, threadIdx.y, threadIdx.z,
+	 blockIdx.x, blockIdx.y, blockIdx.z,
+	 blockDim.x, blockDim.y, blockDim.z,
+	 gridDim.x, gridDim.y,gridDim.z
+	 );
+}
+
+int main(int argc, char** argv){
+  // データの要素の合計数
+  int nElem = 6;
+
+  // グリッドとブロック構造を定義
+  //
+  // thread < block < grid 
+  //
+  // dim3 はdim3宣言とよばれていて
+  // dim3 grid(10,10)はgridの中に10x10のblockを起動する
+  // dim3 block(8,8,8)はblock中に8x8x8のthreadを起動する
+  
+  dim3 block(3); // ブロック中に3 thread を起動する
+  
+  // 指定されたデータサイズとスレッド数からグリッド数の引数(ブロック数)を指定する
+  // (6 + 3 - 1) /  3 =  2 
+  dim3 grid((nElem + block.x - 1)/block.x);
+
+  // グリッドサイズをホスト側から確認
+  printf("grid.x %d grid.y %d grid.z %d\n", grid.x, grid.y, grid.z);
+  printf("block.x %d block.y %d block.z %d\n", block.x, block.y, block.z);
+
+  // グリッドサイズをデバイス側から確認
+  checkIndex<<<grid, block>>>();
+
+  // デイバイスをリセット(これを記述しないと動かない)
+  cudaDeviceReset();
+  return(0);
+}

check_dimension/check_dimension.cu

@@ -2,7 +2,7 @@
 #include <time.h>
 
 // ホスト側でベクトルの和を計算
-void sumArrayOnHost(float *A, float *B, float *C, cont in N){
+void sumArraysOnHost(float *A, float *B, float *C, const int N){
   for(int idx = 0; idx < N; idx++){
     C[idx] = A[idx] + B[idx];
   }
@@ -21,7 +21,22 @@ void initialData(float *ip, int size){
 
 int main(int argc, char** argv){
   int nElem = 1024;
+  size_t nByte = nElem * sizeof(float);
+  // ホスト側の変数にはプレフィックス "h_" を付与する 
+  float *h_A, *h_B, *h_C;
+  h_A = (float *)malloc(nByte);
+  h_B = (float *)malloc(nByte);
+  h_C = (float *)malloc(nByte);
+  // 配列の要素を初期化
+  initialData(h_A, nElem);
+  initialData(h_B, nElem);
 
-  //  工事中...
-  
+  // ホストで配列の和を計算
+  sumArraysOnHost(h_A, h_B, h_C, nElem);
+
+  // メモリを開放
+  free(h_A);
+  free(h_B);
+  free(h_C);
+  return(0);
 }