Skip to content

Commit fc9e955

Browse files
slarenggerganov
andauthored
ggml backends interface v1 (#547)
* ggml backends interface v1 * ggml-backend : metal (#552) --------- Co-authored-by: Georgi Gerganov <[email protected]>
1 parent 4a771d5 commit fc9e955

File tree

14 files changed

+1414
-259
lines changed

14 files changed

+1414
-259
lines changed

.gitignore

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@ build-sanitize-thread/
66
build-cov/
77
build-ci-debug/
88
build-ci-release/
9+
build-cublas/
910
out/
1011
tmp/
1112
models/
@@ -15,6 +16,7 @@ compile_commands.json
1516
CMakeSettings.json
1617
.vs/
1718
.vscode/
19+
.clangd
1820

1921
.exrc
2022
.cache
@@ -32,4 +34,4 @@ zig-cache/
3234

3335
*.sw?
3436

35-
__pycache__/
37+
__pycache__/

examples/gpt-2/CMakeLists.txt

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,3 +11,18 @@ target_link_libraries(${TEST_TARGET} PRIVATE ggml common common-ggml)
1111
set(TEST_TARGET gpt-2-quantize)
1212
add_executable(${TEST_TARGET} quantize.cpp)
1313
target_link_libraries(${TEST_TARGET} PRIVATE ggml common common-ggml)
14+
15+
#
16+
# For GPU offloading
17+
18+
if (GGML_CUBLAS)
19+
add_compile_definitions(GGML_USE_CUBLAS)
20+
endif()
21+
22+
if (GGML_CLBLAST)
23+
add_compile_definitions(GGML_USE_CLBLAST)
24+
endif()
25+
26+
if (GGML_METAL)
27+
add_compile_definitions(GGML_USE_METAL)
28+
endif()

0 commit comments

Comments
 (0)