Skip to content

Mhc optimization: Rewrites einsums for better xla fusion - #4664

Open
h1t35h wants to merge 1 commit into
AI-Hypercomputer:mainfrom
h1t35h:mhc-optimization
Open

Mhc optimization: Rewrites einsums for better xla fusion#4664
h1t35h wants to merge 1 commit into
AI-Hypercomputer:mainfrom
h1t35h:mhc-optimization

Manually replacing einsums leading to XLA doing a better optimization…

6bc82c0
Select commit
Loading
Failed to load commit list.
Google CLA / cla/google succeeded Jul 29, 2026 in 14s

✅ All contributors are covered under a CLA with Google

See https://cla.developers.google.com/ for more info about Google's Contributor License Agreement (CLA).

ℹ️ Googlers: Go here to view more details and manage scans for this pull request.

Details

The following contributors were found for this pull request:

6bc82c0 Author: @h1t35h <hit*******v04​@gmail.com>, <hi***y​@google.com>

(Only the first commit for a unique contributor is listed.)