DLNetBench/Makefile.BALDO at main · HicrestLaboratory/DLNetBench · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
CXX := nvcc

# -- System paths (NVHPC 24.7 CUDA 12.5) --
NVHPC_ROOT := /opt/shares/NVHPC/nvhpc_24.7_cuda_12.5/Linux_x86_64/24.7

# -- System paths --
CUDA_HOME := $(NVHPC_ROOT)/cuda
MPI_HOME  := $(NVHPC_ROOT)/comm_libs/12.5/hpcx/hpcx-2.19/ompi
NCCL_HOME := $(NVHPC_ROOT)/install_components/Linux_x86_64/24.7/REDIST/comm_libs/12.5/nccl

# OpenMPI bundled with NVHPC is CUDA-aware
MPICUDA_HOME := $(MPI_HOME)

# -------------------------------------------------------
# Config: nccl
#   CUDA buffers + NCCL collectives
# -------------------------------------------------------
CONFIGS += nccl
BACKEND_nccl := nccl
EXTRA_CXXFLAGS_nccl := -arch=sm_80

# -------------------------------------------------------
# Config: mpi_gpu_cuda
#   CUDA buffers, MPI collectives only (no NCCL)
# -------------------------------------------------------
CONFIGS += mpi_gpu_cuda
BACKEND_mpi_gpu_cuda := mpi_gpu_cuda
EXTRA_CXXFLAGS_mpi_gpu_cuda := -arch=sm_80

# -------------------------------------------------------
# Includes
# -------------------------------------------------------
include Makefile.flags.mk
include Makefile.common