FEAT: Added the ability to make broadcast a decorator

Author: pavanky

arrayfire/broadcast.py

@@ -20,7 +20,14 @@ def toggle():
         bcast._flag ^= True
 
 def broadcast(func, *args):
-    bcast.toggle()
-    res = func(*args)
-    bcast.toggle()
-    return res
+
+    def wrapper(*func_args):
+        bcast.toggle()
+        res = func(*func_args)
+        bcast.toggle()
+        return res
+
+    if len(args) == 0:
+        return wrapper
+    else:
+        return wrapper(*args)

tests/simple_arith.py

@@ -190,8 +190,14 @@
 af.display(af.isnan(a/a))
 
 a = af.randu(5, 1)
-b = af.randu(5, 5)
+b = af.randu(1, 5)
 c = af.broadcast(lambda x,y: x+y, a, b)
 af.display(a)
 af.display(b)
 af.display(c)
+
+@af.broadcast
+def test_add(aa, bb):
+    return aa + bb
+
+af.display(test_add(a, b))