Example #1
0
 static void initDataArray(af_array *arr, const T *ptr, af_source_t src,
                           dim_type d0, dim_type d1=1, dim_type d2=1, dim_type d3=1)
 {
     af::dtype ty = (af::dtype)dtype_traits<T>::af_type;
     dim_type my_dims[] = {d0, d1, d2, d3};
     switch (src) {
     case afHost:   AF_THROW(af_create_array(arr, (const void * const)ptr, 4, my_dims, ty)); break;
     case afDevice: AF_THROW(af_device_array(arr, (const void *      )ptr, 4, my_dims, ty)); break;
     default: AF_THROW(AF_ERR_INVALID_ARG);
     }
 }
Example #2
0
 static void initDataArray(af_array *arr, const T *ptr, af::source src,
                           dim_t d0, dim_t d1=1, dim_t d2=1, dim_t d3=1)
 {
     af::dtype ty = (af::dtype)dtype_traits<T>::af_type;
     dim_t my_dims[] = {d0, d1, d2, d3};
     switch (src) {
     case afHost:   AF_THROW(af_create_array(arr, (const void * const)ptr, AF_MAX_DIMS, my_dims, ty)); break;
     case afDevice: AF_THROW(af_device_array(arr, (const void *      )ptr, AF_MAX_DIMS, my_dims, ty)); break;
     default: AF_THROW_ERR("Can not create array from the requested source pointer",
                           AF_ERR_ARG);
     }
 }