From 7beff72f5f5fb7449c50ef7e4ee6b065643becbc Mon Sep 17 00:00:00 2001
From: ademeure <arundemeure@gmail.com>
Date: Wed, 1 May 2024 15:31:30 +0100
Subject: [PATCH] Add makefile changes for cuDNN

---
 Makefile | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

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)