contract.hpp

#include "./interface/cutensor_interface.hpp"
#include "./interface/tblis_interface.hpp"
#include "./tools.hpp"
#include "../mem/address_space.hpp"
#include "../traits.hpp"
#include <string_view>
#include <utility>

Detailed Description

Provides a generic tensor contraction with cuTENSOR/TBLIS dispatch.

Definition in file contract.hpp.

Go to the source code of this file.

Functions
template<BlasArrayOrConj A, BlasArrayOrConjFor< A > B, BlasArrayFor< A > C>
void	nda::tensor::contract (A const &a, std::string_view idx_a, B const &b, std::string_view idx_b, C &&c, std::string_view idx_c)
	Convenience overload of nda::tensor::contract with \( \alpha = 1 \) and \( \beta = 0 \).
template<BlasArrayOrConj A, BlasArrayOrConjFor< A > B, BlasArrayFor< A > C>
void	nda::tensor::contract (get_value_t< A > alpha, A const &a, std::string_view idx_a, B const &b, std::string_view idx_b, get_value_t< A > beta, C &&c, std::string_view idx_c)
	Tensor contraction with cuTENSOR/TBLIS dispatch.