未验证 提交 502d7daf 编写于 作者: 武毅 提交者: GitHub

Merge pull request #5467 from typhoonzero/fix_accuracy_cudamemset

fix accuracy cudamemset
...@@ -65,7 +65,7 @@ class AccuracyOpCUDAKernel : public framework::OpKernel<T> { ...@@ -65,7 +65,7 @@ class AccuracyOpCUDAKernel : public framework::OpKernel<T> {
size_t num_samples = inference->dims()[0]; size_t num_samples = inference->dims()[0];
size_t infer_width = inference->dims()[1]; size_t infer_width = inference->dims()[1];
cudaMemset((void**)&accuracy_data, 0, sizeof(float)); PADDLE_ENFORCE(cudaMemset(accuracy_data, 0, sizeof(float)));
if (num_samples == 0) { if (num_samples == 0) {
return; return;
......
...@@ -26,5 +26,4 @@ class TestAccuracyOp(OpTest): ...@@ -26,5 +26,4 @@ class TestAccuracyOp(OpTest):
if __name__ == '__main__': if __name__ == '__main__':
exit(0)
unittest.main() unittest.main()
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册