Updated XLA Flags for SparseCore #1210

Obliviour · 2025-01-28T20:19:55Z

Description

Update XLA Flags for SparseCore Collective offloading. Add different combinations of SC offloading and organize them clearly.

Ran various tests with the flags to see different SCs enabled.

Before submitting this PR, please make sure (put X in square brackets):

I have performed a self-review of my code.
I have necessary comments in my code, particularly in hard-to-understand areas.
I have run end-to-end tests tests and provided workload links above if applicable.
I have made or will make corresponding changes to the doc if needed.

Obliviour requested review from gobbleturk, khatwanimohit, bvandermoon, vipannalla and RissyRan as code owners January 28, 2025 20:19

gobbleturk approved these changes Jan 29, 2025

View reviewed changes

Obliviour force-pushed the sc_xla branch from e40f4ca to f379203 Compare January 29, 2025 23:56

github-actions bot added the pull ready label Jan 30, 2025

Updated XLA Flags for SparseCore

438697f

Obliviour force-pushed the sc_xla branch from f379203 to 438697f Compare January 30, 2025 19:00

copybara-service bot merged commit 1c3d418 into main Jan 30, 2025
13 of 15 checks passed

copybara-service bot deleted the sc_xla branch January 30, 2025 19:28