Skip to content

Commit

Permalink
DOC: Intel MPI GPU offload update (uxlfoundation#2145)
Browse files Browse the repository at this point in the history
  • Loading branch information
ethanglaser authored Nov 4, 2024
1 parent 13e567b commit 08656de
Showing 1 changed file with 8 additions and 0 deletions.
8 changes: 8 additions & 0 deletions doc/sources/distributed-mode.rst
Original file line number Diff line number Diff line change
Expand Up @@ -28,6 +28,14 @@ via ``impi-devel`` python package) and the ``mpi4py`` python package. If using |
`installed from sources <https://github.com/intel/scikit-learn-intelex/blob/main/INSTALL.md#build-from-sources>`_,
ensure that the spmd_backend is built.

Note that |intelex| now supports GPU offloading to speed up MPI operations. This is supported automatically with
some MPI backends, but in order to use GPU offloading with Intel MPI, set the following environment variable (providing
data on device without this may lead to a runtime error):

::

export I_MPI_OFFLOAD=1

Estimators can be imported from the ``sklearnex.spmd`` module. Data should be distributed across multiple nodes as
desired, and should be transfered to a dpctl or dpnp array before being passed to the estimator. View a full
example of this process in the |intelex| repository, where many examples of our SPMD-supported estimators are
Expand Down

0 comments on commit 08656de

Please sign in to comment.