call mpi

EnzymeAD · Mar 4, 2025 · 82fbc9c · 82fbc9c
1 parent 7c272d0
commit 82fbc9c
Show file tree

Hide file tree

Showing 2 changed files with 118 additions and 367 deletions.
diff --git a/src/enzyme_ad/jax/BUILD b/src/enzyme_ad/jax/BUILD
@@ -466,6 +466,7 @@ cc_library(
         "@llvm-project//mlir:MemRefToLLVM",
         "@llvm-project//mlir:NVGPUToNVVM",
         "@llvm-project//mlir:MathDialect",
+        "@llvm-project//mlir:MPIDialect",
         "@llvm-project//mlir:NVGPUDialect",
         "@llvm-project//mlir:VectorDialect",
         "@llvm-project//mlir:MemRefDialect",