ArrayFire bindings for Linear Algebra

arrayfire.gemspec

@@ -21,4 +21,6 @@ Gem::Specification.new do |gem|
   gem.add_development_dependency 'rake-compiler', '~>0.8'
   gem.add_development_dependency 'rdoc', '~>4.0', '>=4.0.1'
   gem.add_development_dependency "minitest", "~> 5.0"
+
+  gem.add_development_dependency 'nmatrix', '>= 0.2.1'
 end

ext/mri/arrayfire.c

@@ -1,4 +1,3 @@
-#include "arrayfire.h"
 #include <stdio.h>
 #include <math.h>
 
@@ -17,6 +16,9 @@ VALUE Random = Qnil;
 VALUE Sparse = Qnil;
 VALUE Statistics = Qnil;
 VALUE Util = Qnil;
+extern "C++" {
+  VALUE cNMatrix;
+}
 
 // prototypes
 void Init_arrayfire();
@@ -330,6 +332,12 @@ DECL_UNARY_RUBY_ACCESSOR(ceil)
 static VALUE arf_eqeq(VALUE left_val, VALUE right_val);
 static VALUE arf_eqeq_approx(VALUE left_val, VALUE right_val);
 
+// Interfaces
+
+static VALUE arf_af_array_to_nmatrix(VALUE self);
+extern VALUE arf_nmatrix_to_af_array_method(VALUE nmatrix);
+afstruct* arf_nmatrix_to_af_array(VALUE nm);
+
 void Init_arrayfire() {
   ArrayFire = rb_define_module("ArrayFire");
 
@@ -411,6 +419,11 @@ void Init_arrayfire() {
   rb_define_method(Af_Array, "floor", (METHOD)arf_unary_floor, 0);
   rb_define_method(Af_Array, "ceil", (METHOD)arf_unary_ceil, 0);
 
+  rb_define_method(Af_Array, "to_nmatrix", (METHOD)arf_af_array_to_nmatrix, 0);
+
+  cNMatrix = rb_define_class("NMatrix", rb_cObject);
+  rb_define_method(cNMatrix, "to_af_array", (METHOD)arf_nmatrix_to_af_array_method, 0);
+
   Algorithm = rb_define_class_under(ArrayFire, "Algorithm", rb_cObject);
   rb_define_singleton_method(Algorithm, "sum", (METHOD)arf_sum, 2);
   rb_define_singleton_method(Algorithm, "sum_nan", (METHOD)arf_sum_nan, 3);
@@ -753,3 +766,5 @@ static VALUE arf_eqeq_approx(VALUE left_val, VALUE right_val) {
 #include "cmodules/sparse.c"
 #include "cmodules/statistics.c"
 #include "cmodules/util.c"
+
+#include "interfaces/nmatrix.c"

ext/mri/extconf.rb

@@ -2,7 +2,9 @@
 
 extension_name = 'arrayfire'
 
-dir_config(extension_name)
+nmatrix_path = Gem::Specification.find_all_by_name('nmatrix').compact
+abort "Cannot locate NMatrix installation" unless nmatrix_path
+nmatrix_header_dir = File.join(nmatrix_path[0].require_path)
 
 $INSTALLFILES = [
   ['ruby_arrayfire.h'       , '$(archdir)'],
@@ -15,19 +17,36 @@
 $CXXFLAGS = ["-Wall -Werror=return-type",$CXXFLAGS].join(" ")
 $CPPFLAGS = ["-Wall -Werror=return-type",$CPPFLAGS].join(" ")
 
-dir_config('arrayfire', '/usr/local/include/', '/usr/local/lib/')
 
-# $LOCAL_LIBS =
 LIBDIR      = RbConfig::CONFIG['libdir']
 INCLUDEDIR  = RbConfig::CONFIG['includedir']
 
-HEADER_DIRS = [INCLUDEDIR, '/usr/local/include/', '/usr/local/include/af/']
+HEADER_DIRS = [
+  '/opt/local/include',
+  '/usr/local/include',
+  INCLUDEDIR,
+  '/usr/include',
+  nmatrix_header_dir
+]
+
+LIB_DIRS = [
+  '/opt/local/lib',
+  '/usr/local/lib',
+  LIBDIR,
+  '/usr/lib',
+  nmatrix_header_dir
+]
+
+dir_config(extension_name, HEADER_DIRS, LIB_DIRS)
 
-have_library('afopencl')
+have_library('afcuda')
 have_library('cusolver')
 have_library('cudart')
 have_library('cufft')
 have_library('cublas')
+have_library('nmatrix')
+have_header("nmatrix_config.h")
+abort "Cannot locate NMatrix header files : nmatrix.h" unless find_header("nmatrix.h")
 
 basenames = %w{ruby_arrayfire}
 $objs = basenames.map { |b| "#{b}.o"   }

ext/mri/interfaces/nmatrix.c

@@ -0,0 +1,58 @@
+static VALUE arf_af_array_to_nmatrix(VALUE self) {
+  afstruct* input;
+  Data_Get_Struct(self, afstruct, input);
+  dim_t count;
+  uint ndims;
+  af_get_numdims(&ndims, input->carray);
+
+  dim_t* dims = (dim_t*)malloc(ndims * sizeof(dim_t));
+
+  af_get_dims(&dims[0], &dims[1], &dims[2], &dims[3], input->carray);
+
+  size_t* shape = (size_t*)malloc(ndims * sizeof(size_t));;
+  for (dim_t index = 0; index < ndims; index++){
+    shape[index] = (size_t)(dims[index]);
+  }
+
+  af_get_elements(&count, input->carray);
+
+  double* elements = (double*)malloc(count * sizeof(double));
+  af_get_data_ptr(elements, input->carray);
+
+  return rb_nmatrix_dense_create(nm::FLOAT64, shape, ndims, elements, (int)count);
+}
+
+extern VALUE arf_nmatrix_to_af_array_method(VALUE nmatrix) {
+  if (NM_DIM(nmatrix) > 4) {
+    rb_raise(rb_eStandardError,
+      "NMatrix must not have greater than 4 dimensions.");
+  }
+
+  if (NM_DTYPE(nmatrix) == nm::FLOAT64) {
+    return Data_Wrap_Struct(Af_Array, NULL, arf_free, arf_nmatrix_to_af_array(nmatrix));
+  }
+  else {
+    rb_raise(rb_eStandardError,
+      "NMatrix should be either :complex64, :complex128, :int32 or :float64 type.");
+  }
+  return Qnil;
+}
+
+
+afstruct* arf_nmatrix_to_af_array(VALUE nm) {
+  DENSE_STORAGE* nmat = NM_STORAGE_DENSE(nm);
+  afstruct* output = ALLOC(afstruct);
+
+  if (nmat->dtype != nm::FLOAT64) {
+    rb_raise(rb_eStandardError, "requires dtype of :float64 to convert to an Af_Array");
+  }
+
+  dim_t* shape = (dim_t*)malloc(nmat->dim * sizeof(dim_t));;
+  for (size_t index = 0; index < nmat->dim; index++){
+    shape[index] = (size_t)(nmat->shape[index]);
+  }
+
+  af_create_array(&output->carray, nmat->elements, nmat->dim, shape, f64);
+
+  return output;
+}

ext/mri/ruby_arrayfire.cpp

@@ -7,6 +7,8 @@
 #include <stdio.h>
 #include <math.h>
 #include <map>
+#include "nmatrix.h"
+
 
 /*
  * Project Includes

ext/mri/ruby_arrayfire.h

@@ -2,8 +2,6 @@
   #define RUBY_ARRAYFIRE_H
 #endif
 
-#include <ruby.h>
-
 typedef struct AF_STRUCT
 {
   af_array carray;
@@ -43,6 +41,9 @@ rb_array_const_ptr(VALUE a)
 #ifdef __cplusplus
 typedef VALUE (*METHOD)(...);
 //}; // end of namespace nm
+
+// Interfaces
+
 #endif