Skip to content

ptx-schedule: classify tensor-map mutations as schedule sites - #1215

Open
uurl wants to merge 1 commit into
NVlabs:mainfrom
uurl:feat/ptx-schedule-tensormap-mutation-sites
Open

ptx-schedule: classify tensor-map mutations as schedule sites#1215
uurl wants to merge 1 commit into
NVlabs:mainfrom
uurl:feat/ptx-schedule-tensormap-mutation-sites

Conversation

@uurl

@uurl uurl commented Aug 31, 2026

Copy link
Copy Markdown
Contributor

Closes #1208.

Summary

Classify tensormap.replace.* instructions as schedule-sensitive sites in ptx-schedule.

Tensor-map replacement operations mutate the tensor-map descriptor through the generic proxy. They must be observable by schedule perturbation independently from the fence.proxy.tensormap::* operation that publishes those mutations.

This adds a dedicated SiteKind::TensorMapMutation rather than classifying these instructions as AsyncProxy or OrderedMemory.

Changes

  • Add SiteKind::TensorMapMutation.
  • Classify tensormap.replace.* instructions as TensorMapMutation.
  • Keep fence.proxy.tensormap::* classified as Fence.
  • Add coverage for emitted tensor-map replacement spellings.
  • Add a perturbation test verifying that a tensor-map mutation can be delayed before its publish fence.
  • Document the new site kind in the ptx-schedule README.

Validation

Validated with:

cargo fmt --all -- --check
cargo test -p ptx-schedule tensormap -- --nocapture
cargo test -p ptx-schedule
cargo clippy -p ptx-schedule --all-targets -- -D warnings
RUSTDOCFLAGS="-D warnings" cargo doc --no-deps -p ptx-schedule
git diff --check

Results:

  • 3 tensor-map-specific tests passed.
  • 22 ptx-schedule unit tests passed.
  • 3 CLI integration tests passed.
  • Clippy passed with warnings denied.
  • Rustdoc passed with warnings denied.

The original and perturbed PTX were also assembled successfully with ptxas.

--list-sites reports:

tensormap.replace.*       -> TensorMapMutation
fence.proxy.tensormap::*  -> Fence

@uurl
uurl force-pushed the feat/ptx-schedule-tensormap-mutation-sites branch 2 times, most recently from 18873a2 to c697c43 Compare August 31, 2026 13:31
@uurl uurl changed the title feat(ptx-schedule): classify tensor-map mutations as schedule sites ptx-schedule: classify tensor-map mutations as schedule sites Aug 31, 2026
Signed-off-by: Raul Estrada <raulestradaa@gmail.com>
@uurl
uurl force-pushed the feat/ptx-schedule-tensormap-mutation-sites branch from c697c43 to 7fd3800 Compare August 31, 2026 13:35
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

ptx-schedule: tensormap.replace proxy ordering is unclassified

1 participant