Skip to content

Commit 90b13a9

Browse files
authored
chore: remove debug statements from execute_engine
1 parent 0471f2d commit 90b13a9

File tree

1 file changed

+0
-7
lines changed

1 file changed

+0
-7
lines changed

core/runtime/execute_engine.cpp

Lines changed: 0 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -171,7 +171,6 @@ std::vector<at::Tensor> execute_engine(std::vector<at::Tensor> inputs, c10::intr
171171
gpu_handles.push_back(outputs[pyt_idx].data_ptr());
172172
}
173173
}
174-
std::cout << "========== 3 ===============" << std::endl;
175174
{
176175
std::unique_ptr<torch::autograd::profiler::RecordProfile> enqueue_profiler_guard;
177176
if (compiled_engine->profile_execution) {
@@ -183,19 +182,13 @@ std::vector<at::Tensor> execute_engine(std::vector<at::Tensor> inputs, c10::intr
183182

184183
// nvinfer1::IExecutionContext::enqueue is not thread safe and we need a mutex for it.
185184
std::unique_lock<std::mutex> lock(compiled_engine->mu);
186-
// std::unique_ptr<TRTEngineProfiler> trt_engine_profiler;
187-
// if (compiled_engine->profile_execution) {
188-
// trt_engine_profiler = std::make_unique<TRTEngineProfiler>(compiled_engine->name);
189-
// compiled_engine->exec_ctx->setProfiler(trt_engine_profiler.get());
190-
// }
191185
compiled_engine->exec_ctx->enqueueV2(gpu_handles.data(), stream, nullptr);
192186
if (compiled_engine->profile_execution) {
193187
LOG_INFO(std::endl << *compiled_engine->trt_engine_profiler);
194188
dump_trace(compiled_engine->trt_engine_profile_path, *compiled_engine->trt_engine_profiler);
195189
compiled_engine->dump_engine_layer_info();
196190
}
197191
}
198-
std::cout << "========== 4 ===============" << std::endl;
199192
return outputs;
200193
}
201194

0 commit comments

Comments
 (0)