We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a7c1ed1 commit 0b9abcbCopy full SHA for 0b9abcb
paddle/contrib/inference/paddle_inference_api_anakin_engine.cc
@@ -54,6 +54,7 @@ bool PaddleInferenceAnakinPredictor::Run(
54
LOG(ERROR) << "copy data from CPU to GPU error";
55
return false;
56
}
57
+ cudaStreamSynchronize(NULL);
58
59
60
executor_.prediction();
@@ -76,6 +77,7 @@ bool PaddleInferenceAnakinPredictor::Run(
76
77
LOG(ERROR) << "copy data from GPU to CPU error";
78
79
80
81
82
return true;
83
0 commit comments