From f4f20046c47145c329ae85102daa6cf6bcaad9d6 Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Wed, 5 Jan 2022 20:07:26 +0800
Subject: [PATCH] fix(mgb): fix tensorrt runtimeopr get output var shape bug

GitOrigin-RevId: b830706a891e4859d8e6a0d3db337a2941577168
---
 src/tensorrt/impl/tensorrt_runtime_opr.cpp | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/tensorrt/impl/tensorrt_runtime_opr.cpp b/src/tensorrt/impl/tensorrt_runtime_opr.cpp
index f0849eb0..822116bb 100644
--- a/src/tensorrt/impl/tensorrt_runtime_opr.cpp
+++ b/src/tensorrt/impl/tensorrt_runtime_opr.cpp
@@ -107,6 +107,7 @@ TensorRTRuntimeOpr::TensorRTRuntimeOpr(
 void TensorRTRuntimeOpr::get_output_var_shape(
         const TensorShapeArray& inp_shape, TensorShapeArray& out_shape) const {
     auto batch = inp_shape.at(0)[0];
+    m_manager.clear_trt_context();
     m_manager.create_trt_context(this->comp_node(), inp_shape, m_engine.get());
     auto get_mgb_shape = [&](int binding_idx) -> TensorShape {
         auto dims = m_engine->getBindingDimensions(binding_idx);