build: Add MPI4Py dependency

2024-03-01 23:05:12 +01:00
parent 4f4cdbc10b
commit 044efc6b27
2 changed files with 35 additions and 6 deletions
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -25,9 +25,10 @@ qibo = "^0.2.4"
 quimb = { version = "^1.6.0", extras = ["tensor"] }
 cupy = { version = "^11.6.0", optional = true }
 cuquantum-python-cu11 = { version = "^23.3.0", optional = true }
+mpi4py = { version = "^3.1.5", optional = true }

 [tool.poetry.extras]
-cuda = ["cupy", "cuquantum-python-cu11"]
+cuda = ["cupy", "cuquantum-python-cu11", "mpi4py"]

 [tool.poetry.group.docs]
 optional = true