ciflow/torchtitan/195649: [mtia_compiler][Pytorch] Support stream APIs in torch.compile
- PyTorch: 399 events in the last 90 days
- PyTorch: 396th Release in the last 90 days
- Previous: earlier the same day · ciflow/torchtitan/195645: [INITIAL] Update
What happened
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…
Summary assembled by rule from the sources below