FEAT/TEST: Adding function to call garbage collector

Author: pavanky

arrayfire/device.py

@@ -132,3 +132,9 @@ def device_mem_info():
     mem_info['alloc'] = {'buffers' : alloc_buffers.value, 'bytes' : alloc_bytes.value}
     mem_info['lock'] = {'buffers' : lock_buffers.value, 'bytes' : lock_bytes.value}
     return mem_info
+
+def device_gc():
+    """
+    Ask the garbage collector to free all unlocked memory
+    """
+    safe_call(backend.get().af_device_gc())

tests/simple/device.py

@@ -19,13 +19,17 @@ def simple_device(verbose=False):
     print_func(af.is_dbl_supported())
     af.sync()
 
+    dev = af.get_device()
+    print_func(dev)
     for k in range(af.get_device_count()):
         af.set_device(k)
         dev = af.get_device()
         assert(k == dev)
 
         print_func(af.is_dbl_supported(k))
 
+        af.device_gc()
+
         mem_info_old = af.device_mem_info()
 
         a = af.randu(100, 100)
@@ -34,4 +38,6 @@ def simple_device(verbose=False):
         assert(mem_info['alloc']['buffers'] == 1 + mem_info_old['alloc']['buffers'])
         assert(mem_info[ 'lock']['buffers'] == 1 + mem_info_old[ 'lock']['buffers'])
 
+    af.set_device(dev)
+
 _util.tests['device'] = simple_device