ciflow/torchtitan/195649: [mtia_compiler][Pytorch] Support stream APIs in torch.compile
- PyTorch 近 90 天出现 402 次
- PyTorch 近 90 天第 399 次发版
- 上一次：同一天稍早 · ciflow/torchtitan/195645: [INITIAL] Update
发生了什么
Summary: Register direct torch.mtia stream APIs with Dynamo in-graph rules and special handlers so compiled MTIA functions preserve user-authored stream, current-stream, and synchronization operations. Add an Athena FIE suite covering external and local streams, nested contexts, events and waits, synchronization, and tensor.record_stream() under eager and Inductor backends. Test Plan: buck test fbcode//mtia/host_run…
摘要按规则整理自下方来源原文