diff --git a/Makefile b/Makefile index 73b83720c..db6520bb8 100644 --- a/Makefile +++ b/Makefile @@ -123,6 +123,7 @@ ifeq ($(USE_CUDNN), 1) endif NVCC_INCLUDES += -I$(CUDNN_FRONTEND_PATH) NVCC_LDFLAGS += -lcudnn + NVCC_LDFLAGS += -lnvrtc NVCC_FLAGS += -DENABLE_CUDNN NVCC_CUDNN = $(BUILD_DIR)/cudnn_att.o else