include/utils.hpp

Go to the documentation of this file.

//
// Created by Aman LaChapelle on 5/18/17.
//
// pytorch_inference
// Copyright (c) 2017 Aman LaChapelle
// Full license at pytorch_inference/LICENSE.txt
//

#ifndef PYTORCH_INFERENCE_EXTRACT_NUMPY_HPP
#define PYTORCH_INFERENCE_EXTRACT_NUMPY_HPP

// Python
#include <Python.h>
#define NPY_NO_DEPRECATED_API NPY_1_7_API_VERSION
#include <numpy/arrayobject.h>

// STL
#include <stdexcept>

// ArrayFire
#include <arrayfire.h>

namespace pytorch::internal {

  inline af::array from_numpy(PyArrayObject *array, int ndim, std::vector<int> dims){

    array = PyArray_GETCONTIGUOUS(array);  // make sure it's contiguous (might already be)

    int array_ndim = PyArray_NDIM(array);
    assert(ndim == array_ndim);

    npy_intp *array_dims = PyArray_SHAPE(array);

    for (int i = 0; i < ndim; i++){
      assert(dims[i] == array_dims[i]);  // make sure dimensions are right
    }

    for (int i = dims.size(); i < 4; i++){
      dims.push_back(1);
    }

    int n, k, h, w;
    n = dims[0]; k = dims[1]; h = dims[2]; w = dims[3];

    af::array out (w, h, k, n, reinterpret_cast<float *>(PyArray_DATA(array)));  // errors out here
    out = af::reorder(out, 1, 0, 2, 3);  // reorder to arrayfire specs (h, w, k, batch)

    return out;

  }

  inline void check_size(const int &size1, const int &size2, const std::string &func){
    if (size1 == size2){
      return;
    }

    std::string error = "Incorrect size passed! Sizes: " + std::to_string(size1) + ", " + std::to_string(size2);
    error += " Function: " + func;
    throw std::runtime_error(error);

  }

  inline void check_num_leq(const int &size1, const int &size2, const std::string &func){
    if (size1 <= size2){
      return;
    }

    std::string error = "Incorrect size passed! Sizes: " + std::to_string(size1) + ", " + std::to_string(size2);
    error += " Function: " + func;
    throw std::runtime_error(error);

  }

} // pytorch::internal

#endif //PYTORCH_INFERENCE_EXTRACT_NUMPY_HPP