Set current device to array device before releasing array

umar456 · pradeep · commit 3e038d2e81dc · 2020-06-27T13:08:59.000+05:30
* Arrays were being freed prematurely because the device associated
  with the array was not active. This change makes sure that we call
  setDevice before the memory is freed so we find the correct pointer
  in the memory manager.
diff --git a/src/api/c/handle.hpp b/src/api/c/handle.hpp
@@ -139,7 +139,16 @@ af_array copyArray(const af_array in) {
 
 template<typename T>
 void releaseHandle(const af_array arr) {
-    detail::destroyArray(static_cast<detail::Array<T> *>(arr));
+    auto &Arr      = getArray<T>(arr);
+    int old_device = detail::getActiveDeviceId();
+    int array_id   = Arr.getDevId();
+    if (array_id != old_device) {
+        detail::setDevice(array_id);
+        detail::destroyArray(static_cast<detail::Array<T> *>(arr));
+        detail::setDevice(old_device);
+    } else {
+        detail::destroyArray(static_cast<detail::Array<T> *>(arr));
+    }
 }
 
 template<typename T>
