File tree Expand file tree Collapse file tree
03-H_Multi_GPU_Parallelization
06-H_Overlap_Communication_and_Computation_MPI
10-H_CUDA_Graphs_and_Device-initiated_Communication_with_NVSHMEM
.master/Using_CUDA_Graphs
solutions/Using_CUDA_Graphs Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -6,6 +6,7 @@ CUDA_HOME ?= /usr/local/cuda
66ifndef MPI_HOME
77$(error MPI_HOME is not set)
88endif
9+ _JSCCOURSE_GPU_ARCH? =80
910GENCODE_SM30 := -gencode arch=compute_30,code=sm_30
1011GENCODE_SM35 := -gencode arch=compute_35,code=sm_35
1112GENCODE_SM37 := -gencode arch=compute_37,code=sm_37
@@ -15,7 +16,7 @@ GENCODE_SM60 := -gencode arch=compute_60,code=sm_60
1516GENCODE_SM70 := -gencode arch=compute_70,code=sm_70
1617GENCODE_SM80 := -gencode arch=compute_80,code=sm_80 -gencode arch=compute_80,code=compute_80
1718GENCODE_SM90 := -gencode arch=compute_90,code=sm_90 -gencode arch=compute_90,code=compute_90
18- GENCODE_FLAGS := $(GENCODE_SM80 ) $( GENCODE_SM90 )
19+ GENCODE_FLAGS := $(GENCODE_SM $( _JSCCOURSE_GPU_ARCH ) )
1920ifdef DISABLE_CUB
2021 NVCC_FLAGS = -Xptxas --optimize-float-atomics
2122else
Original file line number Diff line number Diff line change @@ -6,6 +6,7 @@ CUDA_HOME ?= /usr/local/cuda
66ifndef MPI_HOME
77$(error MPI_HOME is not set)
88endif
9+ _JSCCOURSE_GPU_ARCH? =80
910GENCODE_SM30 := -gencode arch=compute_30,code=sm_30
1011GENCODE_SM35 := -gencode arch=compute_35,code=sm_35
1112GENCODE_SM37 := -gencode arch=compute_37,code=sm_37
@@ -15,7 +16,7 @@ GENCODE_SM60 := -gencode arch=compute_60,code=sm_60
1516GENCODE_SM70 := -gencode arch=compute_70,code=sm_70
1617GENCODE_SM80 := -gencode arch=compute_80,code=sm_80 -gencode arch=compute_80,code=compute_80
1718GENCODE_SM90 := -gencode arch=compute_90,code=sm_90 -gencode arch=compute_90,code=compute_90
18- GENCODE_FLAGS := $(GENCODE_SM80 ) $( GENCODE_SM90 )
19+ GENCODE_FLAGS := $(GENCODE_SM $( _JSCCOURSE_GPU_ARCH ) )
1920ifdef DISABLE_CUB
2021 NVCC_FLAGS = -Xptxas --optimize-float-atomics
2122else
Original file line number Diff line number Diff line change @@ -6,6 +6,7 @@ CUDA_HOME ?= /usr/local/cuda
66ifndef MPI_HOME
77$(error MPI_HOME is not set)
88endif
9+ _JSCCOURSE_GPU_ARCH? =80
910GENCODE_SM30 := -gencode arch=compute_30,code=sm_30
1011GENCODE_SM35 := -gencode arch=compute_35,code=sm_35
1112GENCODE_SM37 := -gencode arch=compute_37,code=sm_37
@@ -15,7 +16,7 @@ GENCODE_SM60 := -gencode arch=compute_60,code=sm_60
1516GENCODE_SM70 := -gencode arch=compute_70,code=sm_70
1617GENCODE_SM80 := -gencode arch=compute_80,code=sm_80 -gencode arch=compute_80,code=compute_80
1718GENCODE_SM90 := -gencode arch=compute_90,code=sm_90 -gencode arch=compute_90,code=compute_90
18- GENCODE_FLAGS := $(GENCODE_SM80 ) $( GENCODE_SM90 )
19+ GENCODE_FLAGS := $(GENCODE_SM $( _JSCCOURSE_GPU_ARCH ) )
1920ifdef DISABLE_CUB
2021 NVCC_FLAGS = -Xptxas --optimize-float-atomics
2122else
Original file line number Diff line number Diff line change @@ -6,6 +6,7 @@ CUDA_HOME ?= /usr/local/cuda
66ifndef MPI_HOME
77$(error MPI_HOME is not set)
88endif
9+ _JSCCOURSE_GPU_ARCH? =80
910GENCODE_SM30 := -gencode arch=compute_30,code=sm_30
1011GENCODE_SM35 := -gencode arch=compute_35,code=sm_35
1112GENCODE_SM37 := -gencode arch=compute_37,code=sm_37
@@ -15,7 +16,7 @@ GENCODE_SM60 := -gencode arch=compute_60,code=sm_60
1516GENCODE_SM70 := -gencode arch=compute_70,code=sm_70
1617GENCODE_SM80 := -gencode arch=compute_80,code=sm_80 -gencode arch=compute_80,code=compute_80
1718GENCODE_SM90 := -gencode arch=compute_90,code=sm_90 -gencode arch=compute_90,code=compute_90
18- GENCODE_FLAGS := $(GENCODE_SM80 ) $( GENCODE_SM90 )
19+ GENCODE_FLAGS := $(GENCODE_SM $( _JSCCOURSE_GPU_ARCH ) )
1920ifdef DISABLE_CUB
2021 NVCC_FLAGS = -Xptxas --optimize-float-atomics
2122else
Original file line number Diff line number Diff line change @@ -4,6 +4,7 @@ NVCC=nvcc
44MPICXX =mpicxx
55JSC_SUBMIT_CMD ?= srun --gres=gpu:4 --ntasks-per-node 4
66CUDA_HOME ?= /usr/local/cuda
7+ _JSCCOURSE_GPU_ARCH? =80
78GENCODE_SM30 := -gencode arch=compute_30,code=sm_30
89GENCODE_SM35 := -gencode arch=compute_35,code=sm_35
910GENCODE_SM37 := -gencode arch=compute_37,code=sm_37
@@ -13,7 +14,7 @@ GENCODE_SM60 := -gencode arch=compute_60,code=sm_60
1314GENCODE_SM70 := -gencode arch=compute_70,code=sm_70
1415GENCODE_SM80 := -gencode arch=compute_80,code=sm_80 -gencode arch=compute_80,code=compute_80
1516GENCODE_SM90 := -gencode arch=compute_90,code=sm_90 -gencode arch=compute_90,code=compute_90
16- GENCODE_FLAGS := $(GENCODE_SM80 ) $( GENCODE_SM90 )
17+ GENCODE_FLAGS := $(GENCODE_SM $( _JSCCOURSE_GPU_ARCH ) )
1718ifdef DISABLE_CUB
1819 NVCC_FLAGS = -Xptxas --optimize-float-atomics
1920else
Original file line number Diff line number Diff line change @@ -4,6 +4,7 @@ NVCC=nvcc
44MPICXX =mpicxx
55JSC_SUBMIT_CMD ?= srun --gres=gpu:4 --ntasks-per-node 4
66CUDA_HOME ?= /usr/local/cuda
7+ _JSCCOURSE_GPU_ARCH? =80
78GENCODE_SM30 := -gencode arch=compute_30,code=sm_30
89GENCODE_SM35 := -gencode arch=compute_35,code=sm_35
910GENCODE_SM37 := -gencode arch=compute_37,code=sm_37
@@ -13,7 +14,7 @@ GENCODE_SM60 := -gencode arch=compute_60,code=sm_60
1314GENCODE_SM70 := -gencode arch=compute_70,code=sm_70
1415GENCODE_SM80 := -gencode arch=compute_80,code=sm_80 -gencode arch=compute_80,code=compute_80
1516GENCODE_SM90 := -gencode arch=compute_90,code=sm_90 -gencode arch=compute_90,code=compute_90
16- GENCODE_FLAGS := $(GENCODE_SM80 ) $( GENCODE_SM90 )
17+ GENCODE_FLAGS := $(GENCODE_SM $( _JSCCOURSE_GPU_ARCH ) )
1718ifdef DISABLE_CUB
1819 NVCC_FLAGS = -Xptxas --optimize-float-atomics
1920else
Original file line number Diff line number Diff line change @@ -4,6 +4,7 @@ NVCC=nvcc
44MPICXX =mpicxx
55JSC_SUBMIT_CMD ?= srun --gres=gpu:4 --ntasks-per-node 4
66CUDA_HOME ?= /usr/local/cuda
7+ _JSCCOURSE_GPU_ARCH? =80
78GENCODE_SM30 := -gencode arch=compute_30,code=sm_30
89GENCODE_SM35 := -gencode arch=compute_35,code=sm_35
910GENCODE_SM37 := -gencode arch=compute_37,code=sm_37
@@ -13,7 +14,7 @@ GENCODE_SM60 := -gencode arch=compute_60,code=sm_60
1314GENCODE_SM70 := -gencode arch=compute_70,code=sm_70
1415GENCODE_SM80 := -gencode arch=compute_80,code=sm_80 -gencode arch=compute_80,code=compute_80
1516GENCODE_SM90 := -gencode arch=compute_90,code=sm_90 -gencode arch=compute_90,code=compute_90
16- GENCODE_FLAGS := $(GENCODE_SM80 ) $( GENCODE_SM90 )
17+ GENCODE_FLAGS := $(GENCODE_SM $( _JSCCOURSE_GPU_ARCH ) )
1718ifdef DISABLE_CUB
1819 NVCC_FLAGS = -Xptxas --optimize-float-atomics
1920else
Original file line number Diff line number Diff line change @@ -5,6 +5,7 @@ JSC_SUBMIT_CMD ?= srun --cpu-bind=socket --gres=gpu:4 --ntasks-per-node 4
55MPICXX =mpicxx
66CUDA_HOME ?= /usr/local/cuda
77NCCL_HOME ?= /usr
8+ _JSCCOURSE_GPU_ARCH? =80
89GENCODE_SM30 := -gencode arch=compute_30,code=sm_30
910GENCODE_SM35 := -gencode arch=compute_35,code=sm_35
1011GENCODE_SM37 := -gencode arch=compute_37,code=sm_37
@@ -14,7 +15,7 @@ GENCODE_SM60 := -gencode arch=compute_60,code=sm_60
1415GENCODE_SM70 := -gencode arch=compute_70,code=sm_70
1516GENCODE_SM80 := -gencode arch=compute_80,code=sm_80 -gencode arch=compute_80,code=compute_80
1617GENCODE_SM90 := -gencode arch=compute_90,code=sm_90 -gencode arch=compute_90,code=compute_90
17- GENCODE_FLAGS := $(GENCODE_SM80 ) $( GENCODE_SM90 )
18+ GENCODE_FLAGS := $(GENCODE_SM $( _JSCCOURSE_GPU_ARCH ) )
1819ifdef DISABLE_CUB
1920 NVCC_FLAGS = -Xptxas --optimize-float-atomics
2021else
Original file line number Diff line number Diff line change 1111ifndef MPI_HOME
1212$(error MPI_HOME is not set)
1313endif
14+ _JSCCOURSE_GPU_ARCH? =80
1415GENCODE_SM30 := -gencode arch=compute_30,code=sm_30
1516GENCODE_SM35 := -gencode arch=compute_35,code=sm_35
1617GENCODE_SM37 := -gencode arch=compute_37,code=sm_37
@@ -20,7 +21,7 @@ GENCODE_SM60 := -gencode arch=compute_60,code=sm_60
2021GENCODE_SM70 := -gencode arch=compute_70,code=sm_70
2122GENCODE_SM80 := -gencode arch=compute_80,code=sm_80 -gencode arch=compute_80,code=compute_80
2223GENCODE_SM90 := -gencode arch=compute_90,code=sm_90 -gencode arch=compute_90,code=compute_90
23- GENCODE_FLAGS := $(GENCODE_SM80 ) $( GENCODE_SM90 )
24+ GENCODE_FLAGS := $(GENCODE_SM $( _JSCCOURSE_GPU_ARCH ) )
2425ifdef DISABLE_CUB
2526 NVCC_FLAGS = -Xptxas --optimize-float-atomics
2627else
Original file line number Diff line number Diff line change @@ -5,6 +5,7 @@ JSC_SUBMIT_CMD ?= srun --cpu-bind=socket --gres=gpu:4 --ntasks-per-node 4
55MPICXX =mpicxx
66CUDA_HOME ?= /usr/local/cuda
77NCCL_HOME ?= /usr
8+ _JSCCOURSE_GPU_ARCH? =80
89GENCODE_SM30 := -gencode arch=compute_30,code=sm_30
910GENCODE_SM35 := -gencode arch=compute_35,code=sm_35
1011GENCODE_SM37 := -gencode arch=compute_37,code=sm_37
@@ -14,7 +15,7 @@ GENCODE_SM60 := -gencode arch=compute_60,code=sm_60
1415GENCODE_SM70 := -gencode arch=compute_70,code=sm_70
1516GENCODE_SM80 := -gencode arch=compute_80,code=sm_80 -gencode arch=compute_80,code=compute_80
1617GENCODE_SM90 := -gencode arch=compute_90,code=sm_90 -gencode arch=compute_90,code=compute_90
17- GENCODE_FLAGS := $(GENCODE_SM80 ) $( GENCODE_SM90 )
18+ GENCODE_FLAGS := $(GENCODE_SM $( _JSCCOURSE_GPU_ARCH ) )
1819ifdef DISABLE_CUB
1920 NVCC_FLAGS = -Xptxas --optimize-float-atomics
2021else
You can’t perform that action at this time.
0 commit comments