Skip to content

Navigation Menu

Appearance settings

View all features
- BY COMPANY SIZE
  Enterprises
  Small and medium teams
  Startups
  Nonprofits
- BY USE CASE
  App Modernization
  DevSecOps
  DevOps
  CI/CD
  View all use cases
- BY INDUSTRY
  Healthcare
  Financial services
  Manufacturing
  Government
  View all industries
View all solutions
- EXPLORE BY TOPIC
  AI
  Software Development
  DevOps
  Security
  View all topics
- EXPLORE BY TYPE
  Customer stories
  Events & webinars
  Ebooks & reports
  Business insights
  GitHub Skills
- SUPPORT & SERVICES
  Documentation
  Customer support
  Community forum
  Trust center
  Partners
- COMMUNITY
  GitHub SponsorsFund open source developers
- PROGRAMS
  Security Lab
  Maintainer Community
  Accelerator
  Archive Program
- REPOSITORIES
  Topics
  Trending
  Collections
- ENTERPRISE SOLUTIONS
  Enterprise platformAI-powered developer platform
- AVAILABLE ADD-ONS
  GitHub Advanced SecurityEnterprise-grade security features
  Copilot for BusinessEnterprise-grade AI features
  Premium SupportEnterprise-grade 24/7 support
Pricing

Search code, repositories, users, issues, pull requests...

Search

Clear

Search syntax tips

Provide feedback

We read every piece of feedback, and take your input very seriously.

Include my email address so I can be contacted

Saved searches

Use saved searches to filter your results more quickly

Name

Query

To see all available qualifiers, see our documentation.

Appearance settings

You signed in with another tab or window. Reload to refresh your session. You signed out in another tab or window. Reload to refresh your session. You switched accounts on another tab or window. Reload to refresh your session.

Dismiss alert

EnzymeAD / Enzyme-JAX Public

Notifications You must be signed in to change notification settings
Fork 26
Star 109

Code
Issues 230
Pull requests 96
Actions
Projects
Security
Insights

Additional navigation options

Code
Issues
Pull requests
Actions
Projects
Security
Insights

MPI Ops and lowering passes#1829

Merged

wsmoses merged 104 commits intomainEnzymeAD/Enzyme-JAX:mainfrom

rpl/mpiEnzymeAD/Enzyme-JAX:rpl/mpiCopy head branch name to clipboard

Jan 9, 2026

Conversation Commits104 (104)Checks Files changed

Merged

MPI Ops and lowering passes#1829
wsmoses merged 104 commits intomainfrom
rpl/mpi

Commits

Commits on Jan 9, 2026

Add mpi commrank op
romanlee
committed
Add bazel rules and #includes, now builds
romanlee
committed
Cleanup MPI TypeDefs
romanlee
committed
Skeleton of MPI to LLVM lowering pass
romanlee
committed
MPICommRankOpLowering pass stuff
romanlee
committed
commrankoplowering pass stuff
romanlee
committed
Update EnzymeXLAOps.td LowerEnzymeXLAMPI.cpp
romanlee
committed
Just trying to get something working
romanlee
committed
Revert "Just trying to get something working"
romanlee
committed
Just trying to get something working take 2
romanlee
committed
Remove args to comm_rank for now, will hard code comm=comm world
romanlee
committed
Add some placeholder mpi lit tests
romanlee
committed
Messing around w tests
romanlee
committed
Toy lowering pass
romanlee
committed
TODO figure out what's going on with return success/failure
romanlee
committed
Starting to resemble an actual lowering pass
romanlee
committed
Call the llvm function with enzymexla.jit_call
romanlee
committed
Fix/use the jit call
romanlee
committed
success() probably is behaving correctly, I just probably don't know how to use
romanlee
committed
Declare and use COMM_WORLD and use func arg
romanlee
committed
cleanup
romanlee
committed
Output operand aliasings
romanlee
committed
cleanup
romanlee
committed
Use rewriter.create<>() instead of ::create(rewriter, ...)
romanlee
committed
Style
romanlee
committed
Remove debug print statement
romanlee
committed
Update mpi.mlir test
romanlee
committed
Use stablehlo.constant, add enzymexla.memory_effects attr, update tests
romanlee
committed
Add EnzymeXLAOp for MPI_Comm_size and lowering pass
romanlee
committed
Add MPI Send Op
romanlee
committed
Switch send arg order to correspond with the underlying c api
romanlee
committed
Fix typo
romanlee
committed
Add count arg to Send Op
romanlee
committed
Add Send lowering pass
romanlee
committed
Don't think we need to run enzyme-hlo-opt in mpi tests
romanlee
committed
Cleanup MPI ops
romanlee
committed
Add MPI_Recv Op
romanlee
committed
buf should be an output of Recv, and other various and sundry items
romanlee
committed
Think we need the buf as both an input and output of Recv based on how
romanlee
committed
Half baked Recv lowering pass
romanlee
committed
Finish Recv Op lowering, add test
romanlee
committed
Add Irecv and Wait Ops
romanlee
committed
Add MPI_Wait lowering pass and test
romanlee
committed
Add Irecv lowering pass. Untested
romanlee
committed
Add half a irecv lit test
romanlee
committed
Fix Irecv lowering pass, finish test
romanlee
committed
Add Isend Op
romanlee
committed
Add Isend lowering pass and lit test
romanlee
committed
Add MPI Barrier Op
romanlee
committed
Add Barrier lowering pass
romanlee
committed
Fix Barrier Op, add lit test
romanlee
committed
Add Allreduce Op
romanlee
committed
Add Allreduce lowering pass and lit test
romanlee
committed
Let Comm_rank/size use input/output tensor paradigm
romanlee
committed
Revert "Let Comm_rank/size use input/output tensor paradigm"
romanlee
committed
Let comm_rank take a rank tensor as input
romanlee
committed
Fix name
romanlee
committed
Modify comm_rank lowering pass to take a rank tensor as input
romanlee
committed
Let comm_size take size as input
romanlee
committed
Update comm_size lit test
romanlee
committed
Fix MPI wait lit test, remove irecv to better isolate what we want to
romanlee
committed
Fix send lit test
romanlee
committed
Remove roundtrip test
romanlee
committed
Add irecv + wait lit test
romanlee
committed
COmments
romanlee
committed
Add stringattr to send op
romanlee
committed
Update MPI Send lowering pass and test to use datatype attr
romanlee
committed
Add datatype strattr to recv, isend, irecv, allreduce (and add op
romanlee
committed
Clang format
romanlee
committed
Cleanup unecessary changes
romanlee
committed
Update Ops summaries
romanlee
committed
Remove inrank arg from comm_rank op, temporarily comment out lowering
romanlee
committed
Uncomment comm rank lowering pass
romanlee
committed
Update comm_rank lowering pass and tests to not take any args
romanlee
committed
Fix comm_rank lowering pass
romanlee
committed
Use `mpi.` prefix for ops
romanlee
committed
Update comm_size op and lowering pass to take no args
romanlee
committed
Update comm_size test
romanlee
committed
Remove inrequest arg from mpi isend op
romanlee
committed
Reuncomment isend lowering
romanlee
committed
Update Isend lowering pass
romanlee
committed
Update isend test
romanlee
committed
Remove inrequest from Irecv op, update lowering pass
romanlee
committed
Update irecv tests
romanlee
committed
MPI Send op, replace string attr with enum attr, first step
romanlee
committed
Test: uncomment lapackuploattrget c api export thing
romanlee
committed
Revert "Test: uncomment lapackuploattrget c api export thing"
romanlee
committed
Placeholder MPIDatatypeAttr enum
romanlee
committed
Use new MPIDatatypeAttr in MPI Send
romanlee
committed
Towards an actual MPIDatatypeAttr
romanlee
committed
Frogot to update cpp side
romanlee
committed
Uncomment SendOp lowering pass
romanlee
committed
Update MPISendOp lowering pass and test to use new MPIDatatypeAttr
romanlee
committed
Flesh out MPI_Datatype enum
romanlee
committed
Oops, accidentatlly removed in last commit
romanlee
committed
Forgot to update EnzymeXLA.cpp, do that here
romanlee
committed
Recv, Isend/recv, allreduce: replace StrAttr datatype with EnumAttr
romanlee
committed
Update the set of MPI datatypes
romanlee
committed
Remove comment
romanlee
committed
Update mpi lit tests
romanlee
committed
Use EnumAttr instead of StrAttr for Op in allreduce
romanlee
committed
Update allreduce lit test
romanlee
committed
Mark comm_rank/size as pure
romanlee
committed
Clang formatter
romanlee
committed

Footer

© 2026 GitHub, Inc.

Footer navigation

Terms
Privacy
Security
Status
Community
Docs
Contact

You can’t perform that action at this time.