Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
104 commits
Select commit Hold shift + click to select a range
c5f72cd
Register MPI symbols on load
mofeing Feb 15, 2025
d5eaa2d
ops
mofeing Feb 16, 2025
5f800fe
Update ext/ReactantMPIExt/Overrides.jl
mofeing Feb 16, 2025
5e4a8cd
Update ext/ReactantMPIExt/Overrides.jl
mofeing Feb 22, 2025
215cb1d
register MPI constants
mofeing Feb 28, 2025
4b54755
Fix MPI specializations
mofeing Mar 3, 2025
e7fd20e
fix some symbol registration
mofeing Mar 3, 2025
774982a
refactor MPI Ops
mofeing Mar 3, 2025
d1fe99b
Add functionality for parsing single operations (Julia code)
mofeing Mar 4, 2025
81dd8f2
Update `Ops.comm_rank` to use handwritten MLIR injection
mofeing Mar 4, 2025
2ebb52a
comment
mofeing Mar 4, 2025
11ef645
Update `Ops.comm_size`
mofeing Mar 4, 2025
406fe08
fixes
mofeing Mar 4, 2025
108c679
Refactor `Ops.barrier`
mofeing Mar 4, 2025
bfac727
Refactor to `try_inject_to_top_block!`
mofeing Mar 4, 2025
20fadc2
Refactor MLIR injection
mofeing Mar 5, 2025
2babb79
Refactor MPI constante registration
mofeing Mar 5, 2025
2c8e95c
Fix type inference in `Ops.hlo_call` on empty args
mofeing Mar 5, 2025
a6738f5
Fix MLIR of `Ops.comm_rank`
mofeing Mar 5, 2025
9710e59
Fix MLIR injection C-functions
mofeing Mar 6, 2025
f865e3e
Go back to `Cint` for registering symbols
mofeing Mar 6, 2025
2778f1d
Add `tryinjectop!`
mofeing Mar 6, 2025
eb5da6e
Add `tryinject!`, `inject!` methods
mofeing Mar 6, 2025
5995a7b
Update `comm_rank`
mofeing Mar 6, 2025
32b28ef
Update `mlirOperationInject`, `mlirOperationParse`
mofeing Mar 6, 2025
6e9b1c5
Add `verify` flag to `tryinject!`, `parse(::Operation)`
mofeing Mar 6, 2025
f4acb15
Update `Ops.comm_rank`
mofeing Mar 6, 2025
85a79ed
Update `comm_rank`, `comm_size`, `barrier`, `wait`
mofeing Mar 11, 2025
4e3477f
Implement `Ops.allreduce`
mofeing Mar 11, 2025
9fb0163
Implement `Ops.send`
mofeing Mar 11, 2025
f14dd80
Remove comment
mofeing Mar 11, 2025
14d84e2
Update `Ops.wait`
mofeing Mar 11, 2025
beebfe8
Remove `comm` argument from `Comm_size`, `Barrier` overrides
mofeing Mar 11, 2025
708a574
Fix `Ops.comm_size`
mofeing Mar 11, 2025
da59f4a
Fix `Ops.barrier`
mofeing Mar 11, 2025
b6a9cdf
Fixes and renames
mofeing Mar 11, 2025
c207e02
Override `MPI.Allreduce!`
mofeing Mar 11, 2025
287bd32
Fix conversion of MPI constants to word-size type
mofeing Mar 11, 2025
0c32d65
Comment unused MPI datatypes
mofeing Mar 11, 2025
7eb4107
small fixes
mofeing Mar 11, 2025
19c0eca
Implement `MPI.Recv!`
mofeing Mar 11, 2025
946339c
Test MPI
mofeing Mar 16, 2025
8d2c3d1
Merge branch 'main' into ss/mpi
mofeing Mar 16, 2025
42bf6b2
Update src/mlir/IR/Operation.jl
mofeing Mar 16, 2025
fad1845
Fix `mpiexec` symbol import
mofeing Mar 16, 2025
0359559
Fix typo
mofeing Mar 16, 2025
dd327ec
Init and Finalize on MPI tests
mofeing Mar 16, 2025
e9e657c
Merge branch 'main' into ss/mpi
mofeing May 18, 2025
113b30f
Fix changes introduces in "feat: IR inject functions (#1217)"
mofeing May 18, 2025
15335a9
return errcode for `send` and `recv!`
mofeing Jul 12, 2025
6d89717
Merge branch 'main' into ss/mpi
romanlee Jul 15, 2025
3391df8
Add tests for Send and Recv!
romanlee Jul 16, 2025
89f4ad4
Chipping away at isend
romanlee Jul 18, 2025
b0cdb42
Finish off Ops.isend. Could be working now
romanlee Jul 21, 2025
3c132d4
Add default constructor for TracedRequest
romanlee Jul 21, 2025
32cd922
Clean up mpi.jl tests
romanlee Jul 21, 2025
1bc9ddc
TracedRequest must be mutable struct? Also make TracedRequest optional
romanlee Jul 22, 2025
1218d5c
Scratching together an implementation of make_tracer for TracedRequest
romanlee Aug 14, 2025
626566b
Additions to TracedRequest and associated functions/uses
romanlee Aug 15, 2025
2703138
compile_xla/mlir might be working properly with TracedRequest now
romanlee Aug 18, 2025
0bac4dc
Add create_result for MPI.Request, fix make_tracer
romanlee Aug 28, 2025
a8643c3
Comment these out, for now we'll assume no Request needs to cross the
romanlee Aug 30, 2025
8d2f168
Add Irecv! and friends. Seems to potentially be working
romanlee Aug 30, 2025
56eb8eb
Fix overrides Irecv!
romanlee Sep 1, 2025
d7a764c
Commit debug testing stuff
romanlee Sep 3, 2025
e24e4b7
Merge branch 'main' into ss/mpi
romanlee Sep 3, 2025
72e73fb
Update test-isend-irecv.jl
Sep 4, 2025
f200a6f
Clean up formatting
romanlee Sep 8, 2025
8f12edd
Send/Recv altogether works as long as message size is large enough
romanlee Sep 9, 2025
ff8c4f5
Add TODO
romanlee Sep 9, 2025
8c975bf
Add test where Send and Recv! are compiled together in one function
romanlee Sep 9, 2025
62657ad
Update tests, send/recv-compiled-together test now working
romanlee Sep 10, 2025
43df184
Cleanup testing mess
romanlee Sep 10, 2025
4aa4371
Fix typo in Ops.irecv!(), return error code from Wait
romanlee Sep 10, 2025
bf8eccc
Update isend/irecv tests
romanlee Sep 10, 2025
d1ca937
irecv!: Need to return recvbuf and modify mlir data I think
romanlee Sep 11, 2025
7481fcd
Fix typo in MPI wait
romanlee Sep 16, 2025
0395fb3
Add launcher.jl
romanlee Sep 16, 2025
514a03e
Merge branch 'main' into ss/mpi-new
romanlee Sep 17, 2025
a1b9218
Don't need this until we need to let a Request cross compile barrier
romanlee Sep 17, 2025
5ee8762
Set size of status to max necessary hopefully
romanlee Sep 18, 2025
4c9e2ac
Add Isend/Irecv/Wait unit test
romanlee Sep 18, 2025
81d2d58
Cleanup return vals in tests
romanlee Sep 18, 2025
bf35523
No need to retun error codes from Send/Recv! now that we have this
romanlee Sep 18, 2025
8a8dcf7
No need to return errorcode from wait either
romanlee Sep 18, 2025
1024888
Remove set_default_backend("cpu") from mpi.jl - Not sure that's the
romanlee Sep 18, 2025
0a1258d
Use tryinject instead of inject in wait for MPI_COMM_WORLD
romanlee Sep 18, 2025
c017627
UPdate tests
romanlee Sep 18, 2025
bc8f9c4
Delete debug files
romanlee Sep 18, 2025
4899d6d
Apply suggestions from code review
giordano Sep 19, 2025
a4e159f
Use the already dlopened libmpi
romanlee Sep 22, 2025
c6def59
Remove function convert_julia_type_to_mpi_datatype(), use MPI.Dataype()
romanlee Sep 22, 2025
53112ca
Cleanup/remove some comments
romanlee Sep 22, 2025
d77a4d9
Cleanup some comments
romanlee Sep 22, 2025
508ed6a
set default backend cpu for mpi tests, then set it back
romanlee Sep 23, 2025
8b4631e
Add test for MPI Barrier
romanlee Sep 23, 2025
36b0d26
Remove comment
romanlee Sep 23, 2025
9a312b7
Comment out MPI.Init and MPI.Finalize in Overrides, not implemented yet
romanlee Sep 23, 2025
8ae076b
Merge branch 'main' into ss/mpi
romanlee Sep 23, 2025
e42716e
Merge branch 'main' into ss/mpi
romanlee Sep 24, 2025
6feb05d
Don't register MPI_ERR_RMA_RANGE so CI green since not defined for
romanlee Sep 24, 2025
1d92490
Update ext/ReactantMPIExt/ReactantMPIExt.jl
romanlee Sep 24, 2025
17f8144
Revert "Don't register MPI_ERR_RMA_RANGE so CI green since not define…
romanlee Sep 24, 2025
8a829ef
Update ext/ReactantMPIExt/ReactantMPIExt.jl
avik-pal Sep 24, 2025
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
36 changes: 0 additions & 36 deletions ext/ReactantMPIExt.jl

This file was deleted.

Loading
Loading