BUGFIX: indexing with boolean arrays

Author: pavanky

arrayfire/index.py

@@ -192,10 +192,24 @@ def __init__ (self, idx):
         self.isSeq   = True
 
         if isinstance(idx, BaseArray):
-            self.idx.arr = idx.arr
+
+            arr = ct.c_void_p(0)
+
+            if (Enum_value(idx.dtype()) ==
+                Enum_value(Dtype.b8)):
+                safe_call(backend.get().af_where(ct.pointer(arr), idx.arr))
+            else:
+                safe_call(backend.get().af_retain_array(ct.pointer(arr), idx.arr))
+
+            self.idx.arr = arr
             self.isSeq   = False
         elif isinstance(idx, ParallelRange):
             self.idx.seq = idx
             self.isBatch = True
         else:
             self.idx.seq = Seq(idx)
+
+    # def __del__(self):
+    #     if not self.isSeq:
+    #         arr = self.idx.arr
+    #         backend.get().af_release_array(arr)

tests/simple_index.py

@@ -66,3 +66,10 @@
 for r in rows:
     af.display(r)
     af.display(b[:,r])
+
+a = af.randu(3)
+c = af.randu(3)
+b = af.constant(1,3,dtype=af.Dtype.b8)
+af.display(a)
+a[b] = c
+af.display(a)