btas.h File Reference
#include <limits>
#include <TiledArray/block_range.h>
#include <TiledArray/dense_shape.h>
#include <TiledArray/external/btas.h>
#include <TiledArray/pmap/replicated_pmap.h>
#include <TiledArray/policies/dense_policy.h>
#include <TiledArray/policies/sparse_policy.h>
#include <TiledArray/shape.h>
#include <TiledArray/sparse_shape.h>
#include <TiledArray/tensor.h>
#include <TiledArray/tensor/tensor_map.h>
Include dependency graph for btas.h:
Namespaces | |
TiledArray | |
TiledArray::detail | |
Functions | |
template<typename T , typename Range_ , typename Storage_ , typename Tensor_ > | |
void | TiledArray::btas_subtensor_to_tensor (const btas::Tensor< T, Range_, Storage_ > &src, Tensor_ &dst) |
Copy a block of a btas::Tensor into a TiledArray::Tensor. More... | |
template<typename Tensor_ , typename T , typename Range_ , typename Storage_ > | |
void | TiledArray::tensor_to_btas_subtensor (const Tensor_ &src, btas::Tensor< T, Range_, Storage_ > &dst) |
Copy a block of a btas::Tensor into a TiledArray::Tensor. More... | |
template<typename DistArray_ , typename BTAS_Tensor_ > | |
void | TiledArray::detail::counted_btas_subtensor_to_tensor (const BTAS_Tensor_ *src, DistArray_ *dst, const typename DistArray_::ordinal_type i, madness::AtomicInt *counter) |
template<typename TA_Tensor_ , typename BTAS_Tensor_ > | |
void | TiledArray::detail::counted_tensor_to_btas_subtensor (const TA_Tensor_ &src, BTAS_Tensor_ *dst, madness::AtomicInt *counter) |
Task function for assigning a tensor to an Eigen submatrix. More... | |
template<bool sparse> | |
auto | TiledArray::detail::make_shape (World &world, const TiledArray::TiledRange &trange) |
template<> | |
auto | TiledArray::detail::make_shape< true > (World &world, const TiledArray::TiledRange &trange) |
template<> | |
auto | TiledArray::detail::make_shape< false > (World &, const TiledArray::TiledRange &) |
template<typename DistArray_ , typename T , typename Range , typename Storage > | |
DistArray_ | TiledArray::btas_tensor_to_array (World &world, const TiledArray::TiledRange &trange, const btas::Tensor< T, Range, Storage > &src, bool replicated=false) |
Convert a btas::Tensor object into a TiledArray::DistArray object. More... | |
template<typename Tile , typename Policy , typename Storage = std::vector<typename Tile::value_type>> | |
btas::Tensor< typename Tile::value_type, btas::DEFAULT::range, Storage > | TiledArray::array_to_btas_tensor (const TiledArray::DistArray< Tile, Policy > &src, int target_rank=-1) |
Convert a TiledArray::DistArray object into a btas::Tensor object. More... | |