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

Draft

ajay-mk wants to merge 22 commits into

ajay/feature/umtensor

Commits on May 12, 2026

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

Commits on May 13, 2026

device: drop redundant compile_test_tier1 instantiation probe
ajay-mk
committed
f0c6ec7
device: drop unused UMTensorArg concept; reword scaffolding comments
ajay-mk
committed
a00c93a
chore: update .gitignore to avoide build directories
ajay-mk
committed
1ad8c69
device: update host-device conversion helpers
ajay-mk
committed
05d7e55
device: constrain UMTensor tile ops to numeric element types
ajay-mk
committed
ee31fa9

Commits on May 21, 2026

Commits on May 22, 2026

chore: reformat source files
ajay-mk
committed
7730bb4

Commits on Jun 6, 2026

Merge branch 'master' into ajay/feature/umtensor
ajay-mk
committed
54b09f9