Fix Storage destruction GC tracking (pytorch#94051)

Pull Request resolved: pytorch#94051
Approved by: https://github.com/Skylion007, https://github.com/malfet

Author: albanD

torch/csrc/Storage.cpp

@@ -41,6 +41,12 @@ PyObject* THPStorage_New(c10::intrusive_ptr<c10::StorageImpl> ptr) {
 
 static void THPStorage_subclass_dealloc(PyObject* self) {
   THPStorage* _self = (THPStorage*)self;
+  // Some subclass of StorageBase are GC-tracked objects even
+  // though the base class is not.
+  auto* type = Py_TYPE(self);
+  if (PyType_HasFeature(type, Py_TPFLAGS_HAVE_GC) != 0) {
+    PyObject_GC_UnTrack(self);
+  }
   if (_self->cdata) {
     c10::raw::intrusive_ptr::decref(_self->cdata);
   }