Skip to content

Phase 3 — opt_einsum-style path planner + plan cache reuse #29

@scttfrdmn

Description

@scttfrdmn

Roadmap phase tracker

This issue tracks trntensor's work on Phase 3 of the trnsci roadmap.

See the suite-level roadmap
for the full phase matrix and cross-project dependencies. A reader-friendly
version lives at trnsci.dev/roadmap/.

What this phase means for trntensor:
Multi-tensor contractions have exponentially many possible
orderings; the FLOP cost varies by 10–100×. opt_einsum solves
this on CPU; Phase 3 does it on NKI.

Done means:

  • multi_einsum with optimize="optimal" / "auto" / "greedy" path selection.
  • Contraction plans cached and re-executed without re-planning.
  • Benchmarks on canonical multi-tensor patterns (MP2, CCSD intermediates, THC-compressed evaluations).
  • Integration with trnblas Phase 3 fused kernels for the hot-path contractions.

Coordination

  • Label: phase-3-perf — matches the same label in every sub-project.
  • Cross-project dependencies, if any, are called out in the "Done means"
    list above. Link any PRs / child issues here for tracking.

Close this issue when all "Done means" items are satisfied.

Metadata

Metadata

Assignees

No one assigned

    Labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions