Check if arrayfire can be run immediately after loading the library

Author: pavanky

arrayfire/library.py

@@ -521,34 +521,27 @@ def __init__(self):
             except:
                 pass
 
+        c_dim4 = c_dim_t*4
+        out = ct.c_void_p(0)
+        dims = c_dim4(10, 10, 1, 1)
+
         # Iterate in reverse order of preference
         for name in ('cpu', 'opencl', 'cuda', ''):
             libnames = self.__libname(name)
             for libname in libnames:
                 try:
                     ct.cdll.LoadLibrary(libname)
                     __name = 'unified' if name == '' else name
-                    self.__clibs[__name] = ct.CDLL(libname)
-                    self.__name = __name
+                    clib = ct.CDLL(libname)
+                    self.__clibs[__name] = clib
+                    err = clib.af_randu(ct.pointer(out), 4, ct.pointer(dims), Dtype.f32.value)
+                    if (err == ERR.NONE.value):
+                        self.__name = __name
+                        clib.af_release_array(out)
                     break;
                 except:
                     pass
 
-        c_dim4 = c_dim_t*4
-
-        out = c_dim_t(0)
-        dims = c_dim4(10, 10, 10, 10)
-
-        for key, value in self.__clibs:
-            err = value.af_randu(ct.pointer(out), 4, ct.pointer(dims), 0)
-            if (err == ERR.NONE.value):
-                if (self.__name != key):
-                    self.__name = key
-                break
-            else:
-                self.__name = None
-                pass
-
         if (self.__name is None):
             raise RuntimeError("Could not load any ArrayFire libraries.\n" + more_info_str)