fix a tiny issue with assertion in kernel

jinpeng · May 23, 2014 · 42eaae6 · 42eaae6
1 parent 6672065
commit 42eaae6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/bin/cuda/cwc-bench-runtime.cu b/bin/cuda/cwc-bench-runtime.cu
@@ -132,7 +132,6 @@ extern "C" void cwc_bench_runtime(ccv_convnet_t* convnet, ccv_array_t* categoriz
 	printf("finished backward propagate first convolutional layer on GPU\n");
 	cudaEventDestroy(start);
 	cudaEventDestroy(stop);
-
 	for (i = 0; i < batch; i++)
 	{
 		printf("doing batch %d of %d\n", i + 1, batch);

diff --git a/lib/cuda/cwc_convnet.cu b/lib/cuda/cwc_convnet.cu
@@ -635,8 +635,8 @@ __global__ static void _cwc_kern_convolutional_forward_propagate(const int strid
 		float* filter, const int filter_rows, const int filter_cols, const int count,
 		float* const biases)
 {
-	assert(gridDim.x * partition * filter_per_block == out_rows * count);
-	assert(gridDim.y == out_cols);
+	assert(gridDim.x * partition * filter_per_block == out_cols * count);
+	assert(gridDim.y == out_rows);
 	assert(gridDim.z == partition);
 	extern __shared__ float shared[];
 	float* shared_block = &shared[0];