Skip to content

Commit 0a7e86a

Browse files
committed
fix tensorrt
1 parent 3d4cd2d commit 0a7e86a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

yolox_ros_cpp/yolox_cpp/src/yolox_tensorrt.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -103,7 +103,7 @@ namespace yolox_cpp{
103103

104104
// DMA input batch data to device, infer on the batch asynchronously, and DMA output back to host
105105
CHECK(cudaMemcpyAsync(buffers[this->inputIndex_], input, 3 * this->input_h_ * this->input_w_ * sizeof(float), cudaMemcpyHostToDevice, stream));
106-
context_->enqueue(1, buffers, stream, nullptr);
106+
context_->enqueueV2(buffers, stream, nullptr);
107107
CHECK(cudaMemcpyAsync(output, buffers[this->outputIndex_], this->output_size_ * sizeof(float), cudaMemcpyDeviceToHost, stream));
108108
cudaStreamSynchronize(stream);
109109

0 commit comments

Comments
 (0)