[RUNTIME] Fix memory leakage of TVMByteArray (#4856)
authorLianmin Zheng <lianminzheng@gmail.com>
Tue, 11 Feb 2020 09:58:36 +0000 (01:58 -0800)
committerGitHub <noreply@github.com>
Tue, 11 Feb 2020 09:58:36 +0000 (01:58 -0800)
include/tvm/runtime/packed_func.h

index d3ba2ae..1373f5f 100644 (file)
@@ -875,7 +875,7 @@ class TVMRetValue : public TVMPODValue_ {
   void Clear() {
     if (type_code_ == kTVMNullptr) return;
     switch (type_code_) {
-      case kTVMStr: delete ptr<std::string>(); break;
+      case kTVMStr: case kTVMBytes: delete ptr<std::string>(); break;
       case kTVMPackedFuncHandle: delete ptr<PackedFunc>(); break;
       case kTVMNDArrayHandle: {
         NDArray::FFIDecRef(static_cast<TVMArrayHandle>(value_.v_handle));