Skip to content

Commit bf0c427

Browse files
committed
Add check for cuda_aware_mpi_enabled
Add check for engine="cupy"
1 parent 41d2e7c commit bf0c427

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

pylops_mpi/utils/_mpi.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -247,7 +247,7 @@ def _mpi_calls(comm: MPI.Comm, func: str, *args, engine: Optional[str] = "numpy"
247247
-------
248248
Result of the MPI call
249249
"""
250-
if engine == "cupy":
250+
if engine == "cupy" and deps.cuda_aware_mpi_enabled:
251251
ncp = get_module(engine)
252252
ncp.cuda.Device().synchronize()
253253
mpi_func = getattr(comm, func)

0 commit comments

Comments
 (0)