ggerganov commited on
Commit
d075551
·
1 Parent(s): 7e268a7

cuda : enable CUDA graphs (#0)

Browse files
Files changed (2) hide show
  1. CMakeLists.txt +1 -0
  2. Makefile +1 -1
CMakeLists.txt CHANGED
@@ -393,6 +393,7 @@ if (WHISPER_CUDA)
393
  endif()
394
 
395
  add_compile_definitions(GGML_USE_CUDA)
 
396
 
397
  if (WHISPER_STATIC)
398
  if (WIN32)
 
393
  endif()
394
 
395
  add_compile_definitions(GGML_USE_CUDA)
396
+ add_compile_definitions(GGML_CUDA_USE_GRAPHS)
397
 
398
  if (WHISPER_STATIC)
399
  if (WIN32)
Makefile CHANGED
@@ -295,7 +295,7 @@ ifdef WHISPER_CUDA
295
  endif
296
 
297
  CFLAGS += -DGGML_USE_CUDA -I/usr/local/cuda/include -I/opt/cuda/include -I$(CUDA_PATH)/targets/$(UNAME_M)-linux/include
298
- CXXFLAGS += -DGGML_USE_CUDA -I/usr/local/cuda/include -I/opt/cuda/include -I$(CUDA_PATH)/targets/$(UNAME_M)-linux/include
299
  LDFLAGS += -lcuda -lcublas -lculibos -lcudart -lcublasLt -lcufft -lpthread -ldl -lrt -L/usr/local/cuda/lib64 -L/opt/cuda/lib64 -L$(CUDA_PATH)/targets/$(UNAME_M)-linux/lib -L/usr/lib/wsl/lib
300
  WHISPER_OBJ += ggml-cuda.o whisper-mel-cuda.o
301
  WHISPER_OBJ += $(patsubst %.cu,%.o,$(wildcard ggml-cuda/*.cu))
 
295
  endif
296
 
297
  CFLAGS += -DGGML_USE_CUDA -I/usr/local/cuda/include -I/opt/cuda/include -I$(CUDA_PATH)/targets/$(UNAME_M)-linux/include
298
+ CXXFLAGS += -DGGML_USE_CUDA -I/usr/local/cuda/include -I/opt/cuda/include -I$(CUDA_PATH)/targets/$(UNAME_M)-linux/include -DGGML_CUDA_USE_GRAPHS
299
  LDFLAGS += -lcuda -lcublas -lculibos -lcudart -lcublasLt -lcufft -lpthread -ldl -lrt -L/usr/local/cuda/lib64 -L/opt/cuda/lib64 -L$(CUDA_PATH)/targets/$(UNAME_M)-linux/lib -L/usr/lib/wsl/lib
300
  WHISPER_OBJ += ggml-cuda.o whisper-mel-cuda.o
301
  WHISPER_OBJ += $(patsubst %.cu,%.o,$(wildcard ggml-cuda/*.cu))