0

以下は、私が期待するcudaコードです。「二重解放または破損」エラーが発生します。多くのデバッグの後、問題のある行が「cudaMemcpy(out, out_device...)」であることを特定しました。このエラーが続く理由について、私は完全に唖然としています。助けていただければ幸いです。

void covariance2(float alpha, complex float* in, complex float* out, int dims[5])
{
    int x = dims[0];
    int y = dims[1];
    int z = dims[2];
    int N = dims[3];
    int M = dims[4];

    ComplexFloat* in_device;
    ComplexFloat* out_device;
    int siz = x*y*z*N*(N+1)/2;
    assert(cudaMalloc(&in_device, sizeof(ComplexFloat)*x*y*z*N*M) == cudaSuccess);
    assert(cudaMalloc(&out_device, sizeof(ComplexFloat)*siz) == cudaSuccess);
    assert(cudaMemcpy(in_device, in, sizeof(ComplexFloat)*x*y*z*N*M, cudaMemcpyHostToDevice) == cudaSuccess);
    dim3 numBlocks(x,y,z);
    dim3 numThreads(1);
    size_t sharedMem = N*M + N*(N+1)/2;

    cudaMemcpy(out, out_device, sizeof(ComplexFloat)*siz, cudaMemcpyDeviceToHost);
    cudaFree(in_device);
    cudaFree(out_device);
}

int main()
    {
    int xxx = 2;
    int yyy = 2;
    int zzz = 1;
    int MMM = 7;
    int NNN = 3;
    int dims[5] = { xxx, yyy, zzz, MMM, NNN };
    float alpha = 5.;
    complex float a = 1.314 + 5.42*_Complex_I;
    complex float* in = (complex float*) malloc(sizeof(complex float)*NNN*MMM*xxx*yyy*zzz);
    for (int i = 0; i < xxx*yyy*zzz*MMM*NNN; i++)
        in[i] = a*pow(i,2);
    complex float* out = (complex float*) malloc(sizeof(complex float)*xxx*yyy*zzz*NNN*(NNN+1)/2);
    assert(out);
    covariance2(alpha, in, out, dims);
    for (int i = 0; i < NNN*(NNN+1)/2; i++)
        printf("i = %d, real = %f, imag = %f\n", i, __real__(out[i]), __imag__(out[i]));
    free(out);
    }
4

1 に答える 1

1

covariance2関数内には次のものがあります。

int N = dims[3];
int M = dims[4];

しかし、main()あなたの中には次のものがあります:

int dims[5] = { xxx, yyy, zzz, MMM, NNN };

M は N の前のパラメータであることに注意してください。dims[]

malloc (of out) と cudaMalloc (of out_device) は N に依存するため、N に 2 つの異なる値 (3 と 7) を使用してサイズを計算します。これはout、 とout_deviceが同じサイズではないことを意味します。これはあなたの意図ではないと思います。(size of)outは N の 3 で計算され、(size of)out_deviceは N の 7 で計算されたので、大きなデータ構造を小さなデータ構造に cudaMemcpy しようとしています。修正は、おそらく covariance2 で次のようなものです。

int M = dims[3];
int N = dims[4];
于 2012-12-21T14:51:51.970 に答える