scale.hpp

#include "./interface/cutensor_interface.hpp"
#include "./interface/tblis_interface.hpp"
#include "./tools.hpp"
#include "../exceptions.hpp"
#include "../mapped_functions.hpp"
#include "../mapped_functions.hxx"
#include "../mem/address_space.hpp"
#include "../traits.hpp"

Detailed Description

Provides a generic in-place tensor scaling with cuTENSOR/TBLIS/nda dispatch.

Definition in file scale.hpp.

Go to the source code of this file.

Functions
template<BlasArray A>
void	nda::tensor::scale (get_value_t< A > alpha, A &&a, unary_op op=unary_op::IDENTITY)
	In-place tensor scaling with cuTENSOR/TBLIS/nda dispatch and optional element-wise unary operation.