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

31c2f979 · Lianmin Zheng · GitHub · 902e21bd · 31c2f979
Unverified Commit 31c2f979 authored Feb 11, 2020 by Lianmin Zheng Committed by GitHub Feb 11, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 additions and 1 deletions

include/tvm/runtime/packed_func.h
+1 -1

No files found.
--- 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));