nm_mpi_ext.h File Reference

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Macros
#define	MPIX_CUDA_AWARE_SUPPORT   1
#define	MPIX_ROCM_AWARE_SUPPORT   1
	returns 1 if there is CUDA aware support, 0 otherwise.
#define	MPIX_GPU_SUPPORT_CUDA   2
	GPU aware for nvidia CUDA.
#define	MPIX_GPU_SUPPORT_HIP   3
	GPU aware for AMD HIP.
#define	MPIX_GPU_SUPPORT_ZE   4
	GPU aware for Intel OneAPI level Zero.

Functions
int	MPIX_Query_cuda_support (void)
	returns 1 if there is rocm aware support, 0 otherwise.
int	MPIX_Query_rocm_support (void)
int	MPIX_GPU_query_support (int gpu_type, int *is_supported)

Macro Definition Documentation

MPIX_CUDA_AWARE_SUPPORT

#define MPIX_CUDA_AWARE_SUPPORT   1

Definition at line 22 of file nm_mpi_ext.h.

MPIX_GPU_SUPPORT_CUDA

#define MPIX_GPU_SUPPORT_CUDA   2

GPU aware for nvidia CUDA.

Definition at line 36 of file nm_mpi_ext.h.

MPIX_GPU_SUPPORT_HIP

#define MPIX_GPU_SUPPORT_HIP   3

GPU aware for AMD HIP.

Definition at line 39 of file nm_mpi_ext.h.

MPIX_GPU_SUPPORT_ZE

#define MPIX_GPU_SUPPORT_ZE   4

GPU aware for Intel OneAPI level Zero.

Definition at line 42 of file nm_mpi_ext.h.

MPIX_ROCM_AWARE_SUPPORT

#define MPIX_ROCM_AWARE_SUPPORT   1

returns 1 if there is CUDA aware support, 0 otherwise.

Definition at line 26 of file nm_mpi_ext.h.

Function Documentation

MPIX_GPU_query_support()

int MPIX_GPU_query_support	(	int	gpu_type,
		int *	is_supported
	)

MPIX_Query_cuda_support()

int MPIX_Query_cuda_support

(

void

)

returns 1 if there is rocm aware support, 0 otherwise.

MPIX_Query_rocm_support()

int MPIX_Query_rocm_support

(

void

)