Spaces:
Running
Running
cuda : enable CUDA graphs (#0)
Browse files- CMakeLists.txt +1 -0
- 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))
|