We need op record_stream that is widely used in DDP\FSDP - Githubissues

intel / torch-xpu-ops

Apache License 2.0

30 stars 21 forks source link

We need op record_stream that is widely used in DDP\FSDP #1055

Closed guangyey closed 2 weeks ago

guangyey commented 2 weeks ago

🚀 The feature, motivation and pitch

As titled. Could we implement op aten::record_stream?

cc @zhangxiaoli73

zhangxiaoli73 commented 2 weeks ago

PR https://github.com/intel/torch-xpu-ops/pull/1047 to add this support.