c - 2次元配列プロセスの次元とスレッド番号の宣言

Question

まず、私の文法がひどい場合は申し訳ありませんが、異なる次元 (160x320) の 2D 配列の処理に問題があります。

dim3    blocks(DIMX/16,DIMZ/32);
dim3    threads(16,16);  

このコードは正常にコンパイルされましたが、何らかの形で処理されたのは 160x160 のみで、残りの配列はまだゼロです。私はそれを間違っていますか？

#include "cuda.h"
#include "conio.h"
#include <fstream>
#include <sstream>
#include <iostream>
#include <assert.h>
#include "../common/book.h"
#define DIMX 160
#define DIMZ 320
#define PI 3.1415926535897932f
#define dx 1.0
#define dz 1.0
#define dt 0.001
#define samp 500
#define nite 1000


__global__ void txz_kernel(float *txz,float *vz)
{
int x = threadIdx.x + blockIdx.x * blockDim.x;
int y = threadIdx.y + blockIdx.y * blockDim.y;
int offset = x + y * blockDim.x * gridDim.x;
vz[offset]=txz[offset]+vz[offset];
}

int  main( void ) 
{
    float              *txz;
    float              *vz;

        HANDLE_ERROR( cudaMalloc( (void**)&txz, DIMX * DIMZ * sizeof(float)));
        HANDLE_ERROR( cudaMalloc( (void**)&vz, DIMX * DIMZ * sizeof(float)));

        float *tempvz = (float*)malloc( sizeof(float)*(DIMX*DIMZ));    
        float *temptxz = (float*)malloc( sizeof(float)*(DIMX*DIMZ)); 

    for (int i=0; i<DIMX; i++) {
        for (int j=0; j<DIMZ; j++) {
        int ij=DIMX*j + i; 
        tempvz[ij]=0.0;
        temptxz[ij]=100.0;
        }
    }

    for (int i=0; i<DIMX; i++) {
        for (int j=(121); j<DIMZ; j++) {
        int ij=DIMX*j + i;       
        tempvz[ij]=0.0;
        temptxz[ij]=150.0;
        }
    }
            HANDLE_ERROR( cudaMemcpy( vz, tempvz,sizeof(float)*(DIMX*DIMZ),cudaMemcpyHostToDevice ) );
            HANDLE_ERROR( cudaMemcpy( txz, temptxz,sizeof(float)*(DIMX*DIMZ),cudaMemcpyHostToDevice ) );
                dim3    blocks(DIMX/16,DIMZ/32);
                dim3    threads(16,16);

            txz_kernel<<<blocks,threads>>>(txz,vz) ;            

            float *tempse = (float*)malloc( sizeof(float)*(DIMX*DIMZ));
            HANDLE_ERROR( cudaMemcpy( tempse, vz,sizeof(float)*(DIMX*DIMZ),cudaMemcpyDeviceToHost ) );
            std::ofstream outseis("contour.ctxt"); // output, normal file
            for (int jj=0; jj<DIMZ; jj++) 
            {
                for (int ii=0; ii<DIMX; ii++) 
                {
                int ij=DIMX*jj + ii;               
                outseis<<tempse[ij]<<" ";   
                }
                outseis<<"\r\n";
            }
}