assign.hpp

#include "./interface/cutensor_interface.hpp"
#include "./interface/tblis_interface.hpp"
#include "./tools.hpp"
#include "../concepts.hpp"
#include "../layout/range.hpp"
#include "../layout_transforms.hpp"
#include "../macros.hpp"
#include "../mem/address_space.hpp"
#include "../traits.hpp"
#include <string_view>
#include <type_traits>
#include <utility>

Detailed Description

Provides a generic tensor assignment with cuTENSOR/TBLIS/nda dispatch.

Definition in file assign.hpp.

Go to the source code of this file.

Functions
template<MemoryArray A, MemoryArray B> requires (have_same_value_type_v<A, B>)
void	nda::tensor::assign (A const &a, B &&b)
	Convenience overload of nda::tensor::assign with nda::tensor::default_index strings.
template<MemoryArray A, MemoryArray B> requires (have_same_value_type_v<A, B>)
void	nda::tensor::assign (A const &a, std::string_view idx_a, B &&b, std::string_view idx_b)
	Tensor assignment with cuTENSOR/TBLIS/nda dispatch.
template<int Axis, int Rank>
decltype(auto)	nda::tensor::detail::slice_axis (auto &&arr, long i)
	Slice arr on the compile-time axis Axis at index i, padding the other axes with range::all.