fix stream

dujw · dujw · commit 61c494f2bc1a · 2022-06-13T09:24:09.000Z
diff --git a/src/application/app_yolo_gpuptr/yolo_gpuptr.cpp b/src/application/app_yolo_gpuptr/yolo_gpuptr.cpp
@@ -324,6 +324,8 @@ namespace YoloGPUPtr{
                 checkCudaRuntime(cudaMalloc(&pimage_device, image_data_size));
                 checkCudaRuntime(cudaMemcpyPeerAsync(pimage_device, tensor->device(), image.device_data, image.device_id, image_data_size, image.stream));
                 checkCudaRuntime(cudaStreamSynchronize(image.stream));
+            }else{
+                tensor->set_stream(image.stream, false);
             }
 
             preprocess_stream = tensor->get_stream();

Original file line number	Diff line number	Diff line change
`@@ -324,6 +324,8 @@ namespace YoloGPUPtr{`
`324`	`324`	`checkCudaRuntime(cudaMalloc(&pimage_device, image_data_size));`
`325`	`325`	`checkCudaRuntime(cudaMemcpyPeerAsync(pimage_device, tensor->device(), image.device_data, image.device_id, image_data_size, image.stream));`
`326`	`326`	`checkCudaRuntime(cudaStreamSynchronize(image.stream));`
	`327`	`+ }else{`
	`328`	`+ tensor->set_stream(image.stream, false);`
`327`	`329`	`}`
`328`	`330`
`329`	`331`	`preprocess_stream = tensor->get_stream();`