Skip to content

Phase 5 — trn2 fused multi-contraction paths #31

@scttfrdmn

Description

@scttfrdmn

Roadmap phase tracker

This issue tracks trntensor's work on Phase 5 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:
Wider PSUM on trn2 lets three-way contractions fuse into a
single kernel that wouldn't fit on trn1.

Done means:

  • trn2-specific fused three-contraction kernel for the DF-MP2 pair-energy → denominator → sum pattern.
  • Runtime detection dispatches to the fused kernel only on trn2.
  • Benchmarks show the fused-path win on trn2 vs the unfused common path on trn1.

Coordination

  • Label: phase-5-generation — 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