Adding blas functions and simple tests

Author: pavanky

arrayfire/__init__.py

@@ -3,3 +3,4 @@
 from .util import *
 from .algorithm import *
 from .device import *
+from .blas import *

arrayfire/algorithm.py

@@ -1,7 +1,6 @@
 from .library import *
 from .array import *
 
-
 def parallel_dim(a, dim, c_func):
     out = array()
     safe_call(c_func(pointer(out.arr), a.arr, c_int(dim)))

arrayfire/blas.py

@@ -0,0 +1,22 @@
+from .library import *
+from .array import *
+
+def matmul(lhs, rhs, lhs_opts=AF_MAT_NONE, rhs_opts=AF_MAT_NONE):
+    out = array()
+    safe_call(clib.af_matmul(pointer(out.arr), lhs.arr, rhs.arr,\
+                             lhs_opts, rhs_opts))
+    return out
+
+def dot(lhs, rhs, lhs_opts=AF_MAT_NONE, rhs_opts=AF_MAT_NONE):
+    out = array()
+    safe_call(clib.af_dot(pointer(out.arr), lhs.arr, rhs.arr,\
+                          lhs_opts, rhs_opts))
+    return out
+
+def transpose(a, conj=False):
+    out = array()
+    safe_call(clib.af_transpose(pointer(out.arr), a.arr, conj))
+    return out
+
+def transpose_inplace(a, conj=False):
+    safe_call(clib.af_transpose_inplace(a.arr, conj))

tests/simple_blas.py

@@ -0,0 +1,17 @@
+#!/usr/bin/python
+import arrayfire as af
+
+a = af.randu(5,5)
+b = af.randu(5,5)
+
+af.print_array(af.matmul(a,b))
+af.print_array(af.matmul(a,b,af.AF_MAT_TRANS))
+af.print_array(af.matmul(a,b,af.AF_MAT_NONE, af.AF_MAT_TRANS))
+
+b = af.randu(5,1)
+af.print_array(af.dot(a,b))
+
+af.print_array(af.transpose(a))
+
+af.transpose_inplace(a)
+af.print_array(a)