aparler/c-llm
1
0
Fork 0
mirror of https://github.com/karpathy/llm.c.git synced 2026-07-27 20:25:09 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add makefile changes for cuDNN

Browse source
This commit is contained in:
ademeure 2024-05-01 15:31:30 +01:00
parent d42ee0d925
commit 7beff72f5f
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
Makefile
View file

@ -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)