Repository navigation

[unified-memory] Mark the write loc physical and check it at every write door (4/7) - #40328

Merged

ch-wan merged 9 commits into

sgl-project:mainfrom

caihuali95:mainline/write-loc-id-space

Oct 2, 2026

Merged

[unified-memory] Mark the write loc physical and check it at every write door (4/7)#40328
ch-wan merged 9 commits into
sgl-project:mainfrom
caihuali95:mainline/write-loc-id-space

Commits on Oct 2, 2026
- feat(unified-memory): declare the write loc's id space explicitly
  
  Caihua Li
  authored and
  ch-wan
  committed
  aed5195
- refactor(unified-memory): stamp the write loc's id space at the producers
  
  Caihua Li
  authored and
  ch-wan
  committed
  c9cbffe
- refactor(unified-memory): mark the write loc physical with a bool
  ch-wan
  committed
  f31e00c
- fix(unified-memory): mark every write loc that can reach a unified pool
  ch-wan
  committed
  b70d103
- fix(unified-memory): check the write loc's mark on every write
  ch-wan
  committed
  914f97f
- fix(kv-cache): unwrap the write loc in every pool write door
  ch-wan
  committed
  2f0b374
- fix(attention): size triton's DCP write mask from the loc tensor
  ch-wan
  committed
  0624e04
- test(unified-memory): give hand-built batches the physical mark
  ch-wan
  committed
  4e49e7e
- docs(unified-memory): say what the physical mark covers
  ch-wan
  committed
  7f8e884