File tree 2 files changed +3
-2
lines changed
2 files changed +3
-2
lines changed Original file line number Diff line number Diff line change @@ -51,8 +51,9 @@ if (LLAMA_BUILD)
51
51
)
52
52
53
53
if (LLAVA_BUILD)
54
- if (LLAMA_CUBLAS)
54
+ if (LLAMA_CUBLAS OR LLAMA_CUDA )
55
55
add_compile_definitions (GGML_USE_CUBLAS)
56
+ add_compile_definitions (GGML_USE_CUDA)
56
57
endif ()
57
58
58
59
if (LLAMA_METAL)
Original file line number Diff line number Diff line change @@ -16,7 +16,7 @@ build.debug:
16
16
CMAKE_ARGS=" -DCMAKE_BUILD_TYPE=Debug" python3 -m pip install --verbose --config-settings=cmake.verbose=true --config-settings=logging.level=INFO --config-settings=install.strip=false --editable .
17
17
18
18
build.cuda :
19
- CMAKE_ARGS=" -DLLAMA_CUBLAS =on" python3 -m pip install --verbose -e .
19
+ CMAKE_ARGS=" -DLLAMA_CUDA =on" python3 -m pip install --verbose -e .
20
20
21
21
build.opencl :
22
22
CMAKE_ARGS=" -DLLAMA_CLBLAST=on" python3 -m pip install --verbose -e .
You can’t perform that action at this time.
0 commit comments