diff --git a/src/runtime/contrib/edgetpu/edgetpu_runtime.h b/src/runtime/contrib/edgetpu/edgetpu_runtime.h index a7a57ff422e3..341062f1c492 100644 --- a/src/runtime/contrib/edgetpu/edgetpu_runtime.h +++ b/src/runtime/contrib/edgetpu/edgetpu_runtime.h @@ -31,6 +31,7 @@ #include #include "../tflite/tflite_runtime.h" +#include "edgetpu.h" namespace tvm { namespace runtime { @@ -43,6 +44,14 @@ namespace runtime { */ class EdgeTPURuntime : public TFLiteRuntime { public: + /*! + * \brief Destructor of EdgeTPURuntime. + * + * NOTE: tflite::Interpreter member should be destruct before the EdgeTpuContext member + * destruction. If the order is reverse, occurs SEGV in the destructor of tflite::Interpreter. + */ + ~EdgeTPURuntime() { interpreter_.reset(); } + /*! * \return The type key of the executor. */