diff --git a/arrayfire/interop.py b/arrayfire/interop.py index e1a5d3b3d..e4cd3561d 100644 --- a/arrayfire/interop.py +++ b/arrayfire/interop.py @@ -27,18 +27,19 @@ AF_NUMPY_FOUND=True - _nptype_to_aftype = {'f4' : Dtype.f32, + _nptype_to_aftype = {'b1' : Dtype.b8, + 'u1' : Dtype.u8, + 'u2' : Dtype.u16, + 'i2' : Dtype.s16, + 's4' : Dtype.u32, + 'i4' : Dtype.s32, + 'f4' : Dtype.f32, + 'c8' : Dtype.c32, + 's8' : Dtype.u64, + 'i8' : Dtype.s64, 'f8' : Dtype.f64, - 'b1' : Dtype.b8, - 'u1' : Dtype.u8, - 'i4' : Dtype.s32, - 's4' : Dtype.u32, - 'i8' : Dtype.s64, - 's8' : Dtype.u64, - 'c8' : Dtype.c32, 'c16' : Dtype.c64} - def np_to_af_array(np_arr): """ Convert numpy.ndarray to arrayfire.Array.