This branch implements tensor contraction between inner tensors and their reduction along an outer tensor's mode.#442

Merged

evaleev merged 87 commits intomasterfrom gaudel/feature/tot_inner_tensor_ops

Apr 6, 2024

+2,485-796

Commits on Feb 5, 2024

Small refactor and addition of inner tensor Hadamard, external, and internal indices in einsum function.

bimalgaudel
authored and
evaleev
committed
Random tensor and tensor-of-tensor generation functions added for unit testing.

bimalgaudel
authored and
evaleev
committed
[wip][skip ci] tensor_contract function for TA::Tensor.

bimalgaudel
authored and
evaleev
committed
Add non-const begin() and end() methods to Einsum::index::Index.

bimalgaudel
authored and
evaleev
committed
Include sstream header.

bimalgaudel
authored and
evaleev
committed
Complete implementation of tensor_contract free function.

bimalgaudel
authored and
evaleev
committed
More useful is_array_v type trait.

bimalgaudel
authored and
evaleev
committed
Less verbose element access.

bimalgaudel
authored and
evaleev
committed
Add 'iterator' typedef to einsum::Index class.

bimalgaudel
authored and
evaleev
committed
.add_to method on TA::Tensor supported when 'this' object is empty.

bimalgaudel
authored and
evaleev
committed
Adds 'tensor_hadamard' free function that supports permutation.

bimalgaudel
authored and
evaleev
committed
Adds .is_identity method to fast check whether the permutation is an identity operation.

bimalgaudel
authored and
evaleev
committed
'einsum' function supports Hadamard+contraction on the outer indices with Hadamard or contraction on the inner indices.

bimalgaudel
authored and
evaleev
committed
Add tests for Hadamard+contraction on outer indices, and contraction on inner indices.

bimalgaudel
authored and
evaleev
committed
Print out manual and einsum result in the failing case.

bimalgaudel
authored and
evaleev
committed
char type is used to construct random tensors.

bimalgaudel
authored and
evaleev
committed
Zero-intialization made explicit in Tensor::add_to.

bimalgaudel
authored and
evaleev
committed
Typo.

bimalgaudel
authored and
evaleev
committed
random_tensor function returns the same TA::Tensor for the same TA::Range. [debug]

bimalgaudel
authored and
evaleev
committed
fence and print [debug].

bimalgaudel
authored and
evaleev
committed
Zero initialize manually computed temp result [debug].

bimalgaudel
authored and
evaleev
committed
Tensor::add_to is the wrong place to zero-initialize because we cannot be certain whether the Tensor object was default constructed or not.

bimalgaudel
authored and
evaleev
committed
Amend previous commit on the basis of second thought :)

bimalgaudel
authored and
evaleev
committed
Restore random_tensor function.

bimalgaudel
authored and
evaleev
committed
Cleanup.

bimalgaudel
authored and
evaleev
committed
Test outer pure-Hadamard with inner tensors contraction.

bimalgaudel
authored and
evaleev
committed
fixed permutation logic in BinaryEngine::init_indices_ + implemented (suboptimal) forms of Mult that were not implemented for the ToT case
evaleev
committed
Range(Permutation perm, Range r) returns a copy of r if perm is null
evaleev
committed
permutation handling in ExprEngines is cleaned up to properly track whether inner/outer permutations are implicit .. this should complete support for contractions and hadamard products for ToT in t…
evaleev
committed
cleanup
evaleev
committed
Tensor(OtherTensor, BipartitePermutation) can handle the case where outer(BipartitePermutation) is null
evaleev
committed
Cleanup.

bimalgaudel
authored and
evaleev
committed
Type traits to get nested ranks and max nested-rank among Tensor and DistArray types.

bimalgaudel
authored and
evaleev
committed
string annotation taken by einsum function can handle tensor-of-tensor annotations.

bimalgaudel
authored and
evaleev
committed
Slow yet correct general tensor product evaluation of arbitrarily nested tensors.

bimalgaudel
authored and
evaleev
committed
Compare 'manual' evaluation against einsum evaluations.

bimalgaudel
authored and
evaleev
committed
Add madness gop fences.

bimalgaudel
authored and
evaleev
committed
[skip ci] typos.

bimalgaudel
authored and
evaleev
committed
[skip ci] add a debug instance.

bimalgaudel
authored and
evaleev
committed
[skip ci] relax tile type restrictions on einsum(std::string const&, ...) function.

bimalgaudel
authored and
evaleev
committed
Restrict to pure Hadamard or contraction, between inner tensors.

bimalgaudel
authored and
evaleev
committed
Tensor times Tensor-of-Tensor can be called with BipartitePermutation in the Tensor::binary function.

bimalgaudel
authored and
evaleev
committed
Add more tests for einsum evaluation between tensors of differing nested ranks.

bimalgaudel
authored and
evaleev
committed
[skip ci] Test case name cleanup.

bimalgaudel
authored and
evaleev
committed
scaling with permutation supported in manual evaluation

bimalgaudel
authored and
evaleev
committed
[skip ci] add more tests

bimalgaudel
authored and
evaleev
committed
pure outer product ToT times T seems correct.

bimalgaudel
authored and
evaleev
committed
Partly restores ef8d203

bimalgaudel
authored and
evaleev
committed
[skip ci] ToT x T contraction support.

bimalgaudel
authored and
evaleev
committed

Commits on Feb 13, 2024

Commits on Feb 18, 2024

[skip ci] Another corner case.
bimalgaudel
committed

Commits on Feb 22, 2024

Commits on Feb 23, 2024

Setup nested-rank reduction evaluations in test fixture [wip]
bimalgaudel
committed

Commits on Feb 24, 2024

Commits on Feb 28, 2024

nested-rank reduced result in DistArray result supported by test fixture.
bimalgaudel
committed

Commits on Feb 29, 2024

[wip] supporting denested result as from a general product of two ToTs.
bimalgaudel
committed

Commits on Mar 5, 2024

Commits on Apr 5, 2024