diff --git a/Makefile b/Makefile index db25491..618933b 100644 --- a/Makefile +++ b/Makefile @@ -14,9 +14,10 @@ CUDA_OUTPUT_FILE = -o $@ # NVCC flags # -t=0 is short for --threads, 0 = number of CPUs on the machine NVCC_FLAGS = -O3 -t=0 --use_fast_math -NVCC_LDFLAGS = -lcublas -lcublasLt -NCLL_INCLUDES = +NVCC_LDFLAGS = -lcublas -lcublasLt -lcudnn +NVCC_INCLUDES = -I../cudnn-frontend/include # TODO: Don't hardcode path NVCC_LDLIBS = +NCLL_INCUDES = ifneq ($(OS), Windows_NT) NVCC := $(shell which nvcc 2>/dev/null)