arma_util.hpp

Go to the documentation of this file.

#ifndef MLPACK_BINDINGS_GO_GONUM_ARMA_UTIL_HPP
#define MLPACK_BINDINGS_GO_GONUM_ARMA_UTIL_HPP

// Include Armadillo via mlpack.
#include <mlpack/core/util/io.hpp>
#include <mlpack/core.hpp>

namespace mlpack {

template<typename T>
inline typename T::elem_type* GetMemory(T& m)
{
  if (m.mem && m.n_elem <= arma::arma_config::mat_prealloc)
  {
    // We need to allocate new memory.
    typename T::elem_type* mem =
        arma::memory::acquire<typename T::elem_type>(m.n_elem);
    arma::arrayops::copy(mem, m.memptr(), m.n_elem);
    return mem;
  }
  else
  {
    arma::access::rw(m.mem_state) = 1;
    // With Armadillo 10 and newer, we must set `n_alloc` to 0 so that
    // Armadillo does not deallocate the memory.
    #if ARMA_VERSION_MAJOR >= 10
      arma::access::rw(m.n_alloc) = 0;
    #endif
    return m.memptr();
  }
}

} // namespace mlpack

#endif