Update build file to include mpi4py cluster.

google · Jun 27, 2024 · f7086cb · f7086cb
1 parent 6235eb3
commit f7086cb
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/jax/BUILD b/jax/BUILD
@@ -952,6 +952,7 @@ pytype_strict_library(
         "_src/clusters/cluster.py",
         "_src/clusters/ompi_cluster.py",
         "_src/clusters/slurm_cluster.py",
+        "_src/clusters/mpi4py_cluster.py",
         "_src/distributed.py",
         "_src/xla_bridge.py",
     ],