search

openxla / xla

A machine learning compiler for GPUs, CPUs, and ML accelerators
Apache License 2.0
2.44k stars 372 forks source link

PR #14202: [fusion] Add RS->DUS dynamic slice fusion #14396

Open copybara-service[bot] opened 2 weeks ago

copybara-service[bot] commented 2 weeks ago

PR #14202: [fusion] Add RS->DUS dynamic slice fusion

Imported from GitHub PR https://github.com/openxla/xla/pull/14202

This patch adds execution support and fusion rewriting support for reduce-scatter -> dynamic-update-slice pattern. Copybara import of the project:

-- 180b1d57405aac7cb60607cb73c7f67d78f16426 by Shraiysh Vaishay svaishay@nvidia.com:

[fusion] Add RS->DUS dynamic slice fusion

This patch adds execution support and fusion rewriting support for reduce-scatter -> dynamic-update-slice pattern.

Merging this change closes #14202

FUTURE_COPYBARA_INTEGRATE_REVIEW=https://github.com/openxla/xla/pull/14202 from shraiysh:rs_ds_fusion 180b1d57405aac7cb60607cb73c7f67d78f16426