diff --git a/paddle/majel/test/cuda_test.cu b/paddle/majel/test/cuda_test.cu index 28cda19ef989483f7658711b9c17f0f7e12e3090..360c2548755458bf5725e9e36ac48728a89ba3e8 100644 --- a/paddle/majel/test/cuda_test.cu +++ b/paddle/majel/test/cuda_test.cu @@ -23,7 +23,7 @@ TEST(Cuda, Equality) { float h_B[10] = { 0.0, 9.0, 8.0, 7.0, 6.0, 5.0, 4.0, 3.0, 2.0, 1.0 }; float h_C[10]; float *d_A, *d_B, *d_C; - + cudaError_t err; // Memory allocation for d_A, d_B and d_C (in the device) err = cudaMalloc((void **) &d_A, sizeof(float)*n); CHECK_ERR(err); @@ -48,9 +48,9 @@ TEST(Cuda, Equality) { err = cudaMemcpy(h_C, d_C, sizeof(float)*n, cudaMemcpyDeviceToHost); CHECK_ERR(err); - EXPECT_EQ(h_C[1], 1.0); - for (size_t i = 1; i < n - 1; ++i) { + EXPECT_EQ(h_C[0], 1.0); + for (int i = 1; i < n - 1; ++i) { EXPECT_EQ(h_C[i], 11.0); } - EXPECT_EQ(h_C[0], 1.0); + EXPECT_EQ(h_C[9], 1.0); }