squall/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Improved formatting of Makefile

Browse Source
This commit is contained in:
akerr 2017-12-05 12:45:06 -08:00
parent d08ba8ac46
commit 4276e46e61
1 changed files with 8 additions and 49 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

57
cutlass_test/Makefile
View File

@ -43,12 +43,6 @@ else
TRANSPOSE := nn TRANSPOSE := nn
endif endif
ifdef deepbench
BENCHMARK_DEEPBENCH := $(deepbench)
else
BENCHMARK_DEEPBENCH := 0
endif
# If defined, GEMMs only compiled with specified alignment restrictions on A and B # If defined, GEMMs only compiled with specified alignment restrictions on A and B
# matrices. Otherwise, kernels are compiled for all feasible alignment options, and # matrices. Otherwise, kernels are compiled for all feasible alignment options, and
# the appropriate kernel is selected. # the appropriate kernel is selected.
@ -70,14 +64,12 @@ INC += -I$(BASE_DIR)
LIBS += -lcublas LIBS += -lcublas
#------------------------------------------------------------------------------- #-------------------------------------------------------------------------------
# Preprocessor definitions # Preprocessor definitions
#------------------------------------------------------------------------------- #-------------------------------------------------------------------------------
ifeq (nt, $(TRANSPOSE)) ifeq (nt, $(TRANSPOSE))
DEFINES += -DTRANSPOSE_B DEFINES += -DTRANSPOSE_B
else ifeq (tn, $(TRANSPOSE)) else ifeq (tn, $(TRANSPOSE))
DEFINES += -DTRANSPOSE_A DEFINES += -DTRANSPOSE_A
@ -93,17 +85,16 @@ NVCCFLAGS += -std=c++11
# Dependency Lists # Dependency Lists
#------------------------------------------------------------------------------- #-------------------------------------------------------------------------------
DEPS := $(call rwildcard, $(BASE_DIR),*.h) \ DEPS := $(call rwildcard, $(BASE_DIR),*.h) \
$(call rwildcard, $(BASE_DIR)cgl,*.h) \ $(call rwildcard, $(BASE_DIR)cgl,*.h) \
$(BASE_DIR)common.mk \ $(BASE_DIR)common.mk \
$(TEST_DIR)Makefile $(TEST_DIR)Makefile
ALL := sgemm \ ALL := sgemm \
dgemm \ dgemm \
hgemm \ hgemm \
igemm igemm
#------------------------------------------------------------------------------- #-------------------------------------------------------------------------------
@ -135,50 +126,30 @@ all : $(ALL)
sgemm: bin/sgemm_$(TRANSPOSE)_$(BIN_SUFFIX) sgemm: bin/sgemm_$(TRANSPOSE)_$(BIN_SUFFIX)
sgemm_testbench: bin/sgemm_testbench_$(BIN_SUFFIX)
bin/sgemm_$(TRANSPOSE)_$(BIN_SUFFIX) : gemm.cu $(DEPS) bin/sgemm_$(TRANSPOSE)_$(BIN_SUFFIX) : gemm.cu $(DEPS)
mkdir -p bin mkdir -p bin
$(NVCC) -DTEST_SGEMM $(DEFINES) $(SM_TARGETS) -o $@ gemm.cu $(NVCCFLAGS) $(CPU_ARCH) $(INC) $(LIBINC) $(LIBS) $(NVCC) -DTEST_SGEMM $(DEFINES) $(SM_TARGETS) -o $@ gemm.cu $(NVCCFLAGS) $(CPU_ARCH) $(INC) $(LIBINC) $(LIBS)
bin/sgemm_testbench_$(BIN_SUFFIX) : gemm_testbench.cu $(DEPS)
mkdir -p bin
$(NVCC) -DTEST_SGEMM $(DEFINES) $(SM_TARGETS) -D BENCHMARK_DEEPBENCH=$(BENCHMARK_DEEPBENCH) -o $@ gemm_testbench.cu $(NVCCFLAGS) $(CPU_ARCH) $(INC) $(LIBINC) $(LIBS)
#------------------------------------------------------------------------------- #-------------------------------------------------------------------------------
# make dgemm # make dgemm
#------------------------------------------------------------------------------- #-------------------------------------------------------------------------------
dgemm: bin/dgemm_$(TRANSPOSE)_$(BIN_SUFFIX) dgemm: bin/dgemm_$(TRANSPOSE)_$(BIN_SUFFIX)
dgemm_testbench: bin/dgemm_testbench_$(BIN_SUFFIX)
bin/dgemm_$(TRANSPOSE)_$(BIN_SUFFIX) : gemm.cu $(DEPS) bin/dgemm_$(TRANSPOSE)_$(BIN_SUFFIX) : gemm.cu $(DEPS)
mkdir -p bin mkdir -p bin
$(NVCC) -DTEST_DGEMM $(DEFINES) $(SM_TARGETS) -o $@ gemm.cu $(NVCCFLAGS) $(CPU_ARCH) $(INC) $(LIBINC) $(LIBS) $(NVCC) -DTEST_DGEMM $(DEFINES) $(SM_TARGETS) -o $@ gemm.cu $(NVCCFLAGS) $(CPU_ARCH) $(INC) $(LIBINC) $(LIBS)
bin/dgemm_testbench_$(BIN_SUFFIX) : gemm_testbench.cu $(DEPS)
mkdir -p bin
$(NVCC) -DTEST_DGEMM $(DEFINES) $(SM_TARGETS) -D BENCHMARK_DEEPBENCH=$(BENCHMARK_DEEPBENCH) -o $@ gemm_testbench.cu $(NVCCFLAGS) $(CPU_ARCH) $(INC) $(LIBINC) $(LIBS)
#------------------------------------------------------------------------------- #-------------------------------------------------------------------------------
# make hgemm # make hgemm
#------------------------------------------------------------------------------- #-------------------------------------------------------------------------------
hgemm: bin/hgemm_$(TRANSPOSE)_$(BIN_SUFFIX) hgemm: bin/hgemm_$(TRANSPOSE)_$(BIN_SUFFIX)
hgemm_testbench: bin/hgemm_testbench_$(BIN_SUFFIX)
bin/hgemm_$(TRANSPOSE)_$(BIN_SUFFIX) : gemm.cu $(DEPS) bin/hgemm_$(TRANSPOSE)_$(BIN_SUFFIX) : gemm.cu $(DEPS)
mkdir -p bin mkdir -p bin
$(NVCC) -DTEST_HGEMM $(DEFINES) $(SM_TARGETS) -o $@ gemm.cu $(NVCCFLAGS) $(CPU_ARCH) $(INC) $(LIBINC) $(LIBS) $(NVCC) -DTEST_HGEMM $(DEFINES) $(SM_TARGETS) -o $@ gemm.cu $(NVCCFLAGS) $(CPU_ARCH) $(INC) $(LIBINC) $(LIBS)
bin/hgemm_testbench_$(BIN_SUFFIX) : gemm_testbench.cu $(DEPS)
mkdir -p bin
$(NVCC) -DTEST_HGEMM $(DEFINES) $(SM_TARGETS) -D BENCHMARK_DEEPBENCH=$(BENCHMARK_DEEPBENCH) -o $@ gemm_testbench.cu $(NVCCFLAGS) $(CPU_ARCH) $(INC) $(LIBINC) $(LIBS)
#------------------------------------------------------------------------------- #-------------------------------------------------------------------------------
# make igemm # make igemm
#------------------------------------------------------------------------------- #-------------------------------------------------------------------------------
@ -189,25 +160,13 @@ bin/igemm_$(TRANSPOSE)_$(BIN_SUFFIX) : gemm.cu $(DEPS)
mkdir -p bin mkdir -p bin
$(NVCC) -DTEST_IGEMM $(DEFINES) $(SM_TARGETS) -o $@ gemm.cu $(NVCCFLAGS) $(CPU_ARCH) $(INC) $(LIBINC) $(LIBS) $(NVCC) -DTEST_IGEMM $(DEFINES) $(SM_TARGETS) -o $@ gemm.cu $(NVCCFLAGS) $(CPU_ARCH) $(INC) $(LIBINC) $(LIBS)
igemm_testbench: bin/igemm_testbench_$(BIN_SUFFIX)
bin/igemm_testbench_$(BIN_SUFFIX) : gemm_testbench.cu $(DEPS)
mkdir -p bin
$(NVCC) -DTEST_IGEMM $(DEFINES) $(SM_TARGETS) -D BENCHMARK_DEEPBENCH=$(BENCHMARK_DEEPBENCH) -o $@ gemm_testbench.cu $(NVCCFLAGS) $(CPU_ARCH) $(INC) $(LIBINC) $(LIBS)
#------------------------------------------------------------------------------- #-------------------------------------------------------------------------------
# make wgemm # make wgemm
#------------------------------------------------------------------------------- #-------------------------------------------------------------------------------
wgemm: bin/wgemm_$(TRANSPOSE)_$(BIN_SUFFIX) wgemm: bin/wgemm_$(TRANSPOSE)_$(BIN_SUFFIX)
wgemm_testbench: bin/wgemm_testbench_$(BIN_SUFFIX)
bin/wgemm_$(TRANSPOSE)_$(BIN_SUFFIX) : gemm.cu $(DEPS) bin/wgemm_$(TRANSPOSE)_$(BIN_SUFFIX) : gemm.cu $(DEPS)
mkdir -p bin mkdir -p bin
$(NVCC) -DTEST_WGEMM -DWMMA $(DEFINES) $(SM_TARGETS) -o $@ gemm.cu $(NVCCFLAGS) $(CPU_ARCH) $(INC) $(LIBINC) $(LIBS) $(NVCC) -DTEST_WGEMM -DWMMA $(DEFINES) $(SM_TARGETS) -o $@ gemm.cu $(NVCCFLAGS) $(CPU_ARCH) $(INC) $(LIBINC) $(LIBS)
bin/wgemm_testbench_$(BIN_SUFFIX) : gemm_testbench.cu $(DEPS)
mkdir -p bin
$(NVCC) -DTEST_WGEMM -DWMMA $(DEFINES) $(SM_TARGETS) -D BENCHMARK_DEEPBENCH=$(BENCHMARK_DEEPBENCH) -o $@ gemm_testbench.cu $(NVCCFLAGS) $(CPU_ARCH) $(INC) $(LIBINC) $(LIBS)