Skip to content

Feat: support for XLA#285

Merged
real-zhangzhe merged 4 commits into
MooreThreads:mainfrom
ChaosAbyssF:xla
Jun 4, 2026
Merged

Feat: support for XLA#285
real-zhangzhe merged 4 commits into
MooreThreads:mainfrom
ChaosAbyssF:xla

Conversation

@ChaosAbyssF

Copy link
Copy Markdown
Contributor

Add XLA registration for several non-tf-native ops:

  • Fusion ops like clip, gelu, layernorm, etc.
  • MusaOneTrans3DEinsum, which differs from native tf einsum op that tf itself provides XLA support.

@real-zhangzhe
real-zhangzhe merged commit ab12158 into MooreThreads:main Jun 4, 2026
39 of 43 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants