fix accuracy cudamemset

paddle/operators/accuracy_op.cu

@@ -14,6 +14,7 @@ limitations under the License. */
 
 #include <thrust/execution_policy.h>
 #include <thrust/reduce.h>
+#include <iostream>
 #include "paddle/operators/accuracy_op.h"
 #include "paddle/platform/cuda_helper.h"
 
@@ -65,7 +66,8 @@ class AccuracyOpCUDAKernel : public framework::OpKernel<T> {
 
     size_t num_samples = inference->dims()[0];
     size_t infer_width = inference->dims()[1];
-    cudaMemset((void**)&accuracy_data, 0, sizeof(float));
+    cudaError_t e = cudaMemset(accuracy_data, 0, sizeof(float));
+    PADDLE_ENFORCE_EQ(0, e, "cudaMemset error");
 
     if (num_samples == 0) {
       return;

python/paddle/v2/framework/tests/test_accuracy_op.py

@@ -26,5 +26,4 @@ class TestAccuracyOp(OpTest):
 
 
 if __name__ == '__main__':
-    exit(0)
     unittest.main()