Einsum, initial PR#2265

Closed

HubertKrawczyk wants to merge 32 commits into

apache:mainfrom

HubertKrawczyk:einsum

Commits on May 18, 2025

Commits on May 19, 2025

Commits on Jun 8, 2025

Commits on Jun 17, 2025

Commits on Jun 18, 2025

better plan generation for einsum, handle more cases, added test that does matrix muls and elemwise muls with sums
HubertKrawczyk
committed

Commits on Jun 30, 2025

Commits on Jul 20, 2025

aB_a contraction works, by transposing matrix, no longer change to codegen is needed, elementwise mutliplication between matrix and vector works
HubertKrawczyk
committed

Commits on Aug 3, 2025

Commits on Aug 9, 2025

refactored plan generation and execution into separate steps; plan is optimal; by default no codegen is used
HubertKrawczyk
committed

Commits on Aug 10, 2025