fix unpinning when tensoradaptor is not available (#4450)

1947d87d · Xin Yao · GitHub · 7a41c126 · 1947d87d
Unverified Commit 1947d87d authored Aug 23, 2022 by Xin Yao Committed by GitHub Aug 23, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

src/runtime/ndarray.cc src/runtime/ndarray.cc +2 -1

No files found.
--- a/src/runtime/ndarray.cc
+++ b/src/runtime/ndarray.cc
@@ -67,7 +67,8 @@ struct NDArray::Internal {
      ptr->mem = nullptr;
    } else if (ptr->dl_tensor.data != nullptr) {
      // if the array is still pinned before freeing, unpin it.
-      UnpinContainer(ptr);
+      if (ptr->pinned_by_dgl_)
+        UnpinContainer(ptr);
      dgl::runtime::DeviceAPI::Get(ptr->dl_tensor.ctx)->FreeDataSpace(
          ptr->dl_tensor.ctx, ptr->dl_tensor.data);
    }