init multiple dtypes

Author: prasunanand

ext/mri/arrayfire.c

@@ -19,6 +19,8 @@ VALUE Util = Qnil;
 
 // prototypes
 void Init_arrayfire();
+
+af_dtype arf_dtype_from_rbsymbol(VALUE sym);
 static VALUE arf_init(int argc, VALUE* argv, VALUE self);
 static VALUE arf_alloc(VALUE klass);
 static void arf_free(afstruct* af);
@@ -598,6 +600,7 @@ VALUE arf_init(int argc, VALUE* argv, VALUE self)
  afstruct* afarray;
  Data_Get_Struct(self, afstruct, afarray);
  if(argc > 0){
+ af_dtype dtype = (argc == 4) ? arf_dtype_from_rbsymbol(argv[3]) : f32;
 
  dim_t ndims = (dim_t)FIX2LONG(argv[0]);
  dim_t* dimensions = (dim_t*)malloc(ndims * sizeof(dim_t));
@@ -611,15 +614,13 @@ VALUE arf_init(int argc, VALUE* argv, VALUE self)
  host_array[index] = (float)NUM2DBL(RARRAY_AREF(argv[2], index));
  }
 
- af_create_array(&afarray->carray, host_array, ndims, dimensions, f32);
-
+ af_create_array(&afarray->carray, host_array, ndims, dimensions, dtype);
  af_print_array(afarray->carray);
  }
 
  return self;
 }
 
-
 static VALUE arf_alloc(VALUE klass)
 {
  /* allocate */
@@ -741,11 +742,12 @@ DEF_UNARY_RUBY_ACCESSOR(ceil, ceil)
 #include "cmodules/backend.c"
 #include "cmodules/blas.c"
 #include "cmodules/cuda.c"
+#include "cmodules/data.c"
+#include "cmodules/defines.c"
 #include "cmodules/device.c"
 #include "cmodules/index.c"
-#include "cmodules/opencl.c"
-#include "cmodules/data.c"
 #include "cmodules/lapack.c"
+#include "cmodules/opencl.c"
 #include "cmodules/random.c"
 #include "cmodules/statistics.c"
 #include "cmodules/util.c"

ext/mri/cmodules/defines.c

@@ -0,0 +1,27 @@
+const char* const DTYPE_NAMES[ARF_NUM_DTYPES] = {
+ "f32",
+ "c32",
+ "f64",
+ "c64",
+ "b8",
+ "s32",
+ "u32",
+ "u8",
+ "s64",
+ "u64",
+ "s16",
+ "u16"
+};
+
+af_dtype arf_dtype_from_rbsymbol(VALUE sym) {
+ ID sym_id = SYM2ID(sym);
+
+ for (size_t index = 0; index < ARF_NUM_DTYPES; ++index) {
+ if (sym_id == rb_intern(DTYPE_NAMES[index])) {
+ return static_cast<af_dtype>(index);
+ }
+ }
+
+ VALUE str = rb_any_to_s(sym);
+ rb_raise(rb_eArgError, "invalid data type symbol (:%s) specified", RSTRING_PTR(str));
+}

ext/mri/ruby_arrayfire.h

@@ -14,6 +14,8 @@ typedef struct RANDOM_ENGINE_STRUCT
  af_random_engine cengine;
 }afrandomenginestruct;
 
+#define ARF_NUM_DTYPES 12
+
 #ifndef HAVE_RB_ARRAY_CONST_PTR
 static inline const VALUE *
 rb_array_const_ptr(VALUE a)