tkdnn(trt8) runs now minus the detections ,[executionContext.cpp::enqueueInternal::312] Error Code 3: Internal Error (Parameter check failed at: runtime/api/executionContext.cpp::enqueueInternal::312, condition: mDeviceMemorySize == 0 || mExecutionResources->getDeviceMemory() != nullptr - is the error that causes it

2021-09-07 07:20:46 +05:30
parent 504dee5016
commit 9fa116ce4a
2 changed files with 2 additions and 2 deletions
@@ -137,7 +137,6 @@ int Yolo::computeDetections(Yolo::detection *dets, int &ndets, int netw, int net

    if(predictions == nullptr)
        predictions = new dnnType[output_dim.tot()];
-    checkCuda(cudaDeviceSynchronize());
    checkCuda( cudaMemcpy(predictions, dstData, output_dim.tot()*sizeof(dnnType), cudaMemcpyDeviceToHost));

    int lw = output_dim.w;