[RUNTIME] Fix memory leakage of TVMByteArray (apache#4856)

alexwong · Feb 28, 2020 · 0f86db6 · 0f86db6
1 parent 7306114
commit 0f86db6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/include/tvm/runtime/packed_func.h b/include/tvm/runtime/packed_func.h
@@ -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));