Explicit stream annotation: Set ExecutionStreamId based on frontend attribute

openxla / xla

A machine learning compiler for GPUs, CPUs, and ML accelerators

Apache License 2.0

2.72k stars 438 forks source link

Explicit stream annotation: Set ExecutionStreamId based on frontend attribute #19699

Open chaserileyroberts opened 3 days ago

chaserileyroberts commented 3 days ago

This PR picks up the stream annotation frontend attribute on async methods and assigns the matching ExecutionStreamId. We also adds support for emitting async kCall methods into ir_emitter_unnested.cc

This PR is another part of breaking up PR #17982.