More SM90s

Author: AndiH

06-H_Overlap_Communication_and_Computation_MPI/.master/Makefile

@@ -12,7 +12,8 @@ GENCODE_SM52	:= -gencode arch=compute_52,code=sm_52
 GENCODE_SM60 := -gencode arch=compute_60,code=sm_60
 GENCODE_SM70 := -gencode arch=compute_70,code=sm_70
 GENCODE_SM80 := -gencode arch=compute_80,code=sm_80 -gencode arch=compute_80,code=compute_80
-GENCODE_FLAGS:= $(GENCODE_SM70) $(GENCODE_SM80)
+GENCODE_SM90 := -gencode arch=compute_90,code=sm_90 -gencode arch=compute_90,code=compute_90
+GENCODE_FLAGS:= $(GENCODE_SM80) $(GENCODE_SM90)
 ifdef DISABLE_CUB
  NVCC_FLAGS = -Xptxas --optimize-float-atomics
 else

08-H_NCCL_NVSHMEM/.master/NCCL/Makefile

@@ -13,7 +13,8 @@ GENCODE_SM52	:= -gencode arch=compute_52,code=sm_52
 GENCODE_SM60 := -gencode arch=compute_60,code=sm_60
 GENCODE_SM70 := -gencode arch=compute_70,code=sm_70
 GENCODE_SM80 := -gencode arch=compute_80,code=sm_80 -gencode arch=compute_80,code=compute_80
-GENCODE_FLAGS:= $(GENCODE_SM70) $(GENCODE_SM80)
+GENCODE_SM90 := -gencode arch=compute_90,code=sm_90 -gencode arch=compute_90,code=compute_90
+GENCODE_FLAGS:= $(GENCODE_SM80) $(GENCODE_SM90)
 ifdef DISABLE_CUB
  NVCC_FLAGS = -Xptxas --optimize-float-atomics
 else

08-H_NCCL_NVSHMEM/.master/NVSHMEM/Makefile

@@ -19,7 +19,8 @@ GENCODE_SM52	:= -gencode arch=compute_52,code=sm_52
 GENCODE_SM60 := -gencode arch=compute_60,code=sm_60
 GENCODE_SM70 := -gencode arch=compute_70,code=sm_70
 GENCODE_SM80 := -gencode arch=compute_80,code=sm_80 -gencode arch=compute_80,code=compute_80
-GENCODE_FLAGS:= $(GENCODE_SM70) $(GENCODE_SM80)
+GENCODE_SM90 := -gencode arch=compute_90,code=sm_90 -gencode arch=compute_90,code=compute_90
+GENCODE_FLAGS:= $(GENCODE_SM80) $(GENCODE_SM90)
 ifdef DISABLE_CUB
  NVCC_FLAGS = -Xptxas --optimize-float-atomics
 else