Device support for `TA::Tensor` via `TA::UMTensor`#554

Draft

ajay-mk wants to merge 21 commits into

ajay/feature/umtensor

Commits on May 12, 2026

device: scaffold UMTensor type + is_device_tile specialization
ajay-mk
committed
device: tier-1 tile-op overloads for UMTensor
ajay-mk
committed
device: add permute / shift / mult + perm-variants for UMTensor
ajay-mk
committed
device: fix in-place op dispatch + correct scale semantics, add expression tests
ajay-mk
committed
device: expand UMTensor expression tests (in-place ops, blocks, einsum, contraction variants)
ajay-mk
committed
device: add UMTensor archive support + array-level helpers
ajay-mk
committed
device: explicit instantiations of UMTensor for standard numeric types
ajay-mk
committed
device: add UMTensor dense + vector example programs
ajay-mk
committed

Commits on May 21, 2026

Commits on May 22, 2026

chore: reformat source files
ajay-mk
committed