fix byte array leak (#1679)

Co-authored-by: Chao Ma <mctt90@gmail.com>

fix byte array leak (#1679)
Co-authored-by: Chao Ma <mctt90@gmail.com>
90a103e7 · Jinjing Zhou · GitHub · 558673e1 · 90a103e7
Unverified Commit 90a103e7 authored Jun 22, 2020 by Jinjing Zhou Committed by GitHub Jun 22, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

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

No files found.
--- a/include/dgl/runtime/packed_func.h
+++ b/include/dgl/runtime/packed_func.h
@@ -779,7 +779,7 @@ class DGLRetValue : public DGLPODValue_ {
  void Clear() {
    if (type_code_ == kNull) return;
    switch (type_code_) {
-      case kStr: delete ptr<std::string>(); break;
+      case kStr: case kBytes: delete ptr<std::string>(); break;
      case kFuncHandle: delete ptr<PackedFunc>(); break;
      case kModuleHandle: delete ptr<Module>(); break;
      case kObjectHandle: delete ptr<std::shared_ptr<Object> >(); break;