diff --git a/nvidia_ascend_tools/nvidia_tools/yolov5/src/inference.cu b/nvidia_ascend_tools/nvidia_tools/yolov5/src/inference.cu
index e894fb8..6db3eea 100644
--- a/nvidia_ascend_tools/nvidia_tools/yolov5/src/inference.cu
+++ b/nvidia_ascend_tools/nvidia_tools/yolov5/src/inference.cu
@@ -113,8 +113,8 @@ void Inference::doInference(IExecutionContext& context, cudaStream_t& stream, vo
 
 void Inference::doInferenceV2(IExecutionContext& context, cudaStream_t& stream, void **buffers, unsigned int outputIndex, float* output, int outputSize, int batchSize) 
 {
-    context.enqueue(batchSize, buffers, stream, nullptr);
-    //context.enqueueV2(buffers, stream, nullptr);  
+//    context.enqueue(batchSize, buffers, stream, nullptr);
+    context.enqueueV2(buffers, stream, nullptr);  
     CUDA_CHECK(cudaMemcpyAsync(output, buffers[outputIndex], batchSize * outputSize * sizeof(float), cudaMemcpyDeviceToHost, stream));  
     cudaStreamSynchronize(stream); 
 }