dot.hpp

#include "./interface/cutensor_interface.hpp"
#include "./interface/tblis_interface.hpp"
#include "./tools.hpp"
#include "../algorithms.hpp"
#include "../basic_array.hpp"
#include "../declarations.hpp"
#include "../layout/policies.hpp"
#include "../mem/address_space.hpp"
#include "../mem/policies.hpp"
#include "../traits.hpp"
#include <string_view>

Detailed Description

Provides a generic full tensor dot product with cuTENSOR/TBLIS/nda dispatch.

Definition in file dot.hpp.

Go to the source code of this file.

Functions
template<BlasArrayOrConj A, BlasArrayOrConjFor< A, get_rank< A > > B>
get_value_t< A >	nda::tensor::dot (A const &a, B const &b)
	Convenience overload of nda::tensor::dot with nda::tensor::default_index strings.
template<BlasArrayOrConj A, BlasArrayOrConjFor< A > B>
get_value_t< A >	nda::tensor::dot (A const &a, std::string_view idx_a, B const &b, std::string_view idx_b)
	Full tensor dot product with cuTENSOR/TBLIS/nda dispatch.