GPU Support
This section provides information about the following GPU devices support in Intel® MPI Library:
These features require the Level-Zero* library to be installed on the nodes. All environment variables have I_MPI_OFFLOAD_* prefix.
Current support is limited to Intel® GPUs only.
I_MPI_OFFLOAD
Set this environment variable to enable all GPU features.
Syntax
I_MPI_OFFLOAD=<
value
>Arguments
Value
| Description
|
|---|---|
0
| Disabled (default value)
|
1
| Auto. Intel MPI expects that libze_loader.so is already loaded and does not load it twice
|
2
| Enabled. Intel MPI loads libze_loader.so
|
Description
Set this environment variable to enable all GPU features such as GPU pinning and GPU buffers support, which give you ability to distribute devices between MPI ranks and provide a pointer of an offloaded memory to MPI functions.
I_MPI_OFFLOAD_LEVEL_ZERO_LIBRARY
Specify the name and full path to the Level-Zero library.
Syntax
I_MPI_OFFLOAD_LEVEL_ZERO_LIBRARY="<path>/<name>"
Arguments
<
Full path to the Level-Zero library.
path
><
Name of the Level-Zero library.
name
>Description
Set this environment variable to specify the name and full path to Level-Zero library. Set this variable if Level-Zero is not located in the default path. Default value:
libze_loader.so
.