Skip to content

Commit a113689

Browse files
authored
ggml : add graph tensor allocator (#2411)
* ggml : add graph tensor allocator * ggml : don't calculate data pointer of unallocated tensors when creating a view with an offset * ggml : refactor ggml_view_Nd into ggml_view_tensor_offset
1 parent 11f3ca0 commit a113689

File tree

7 files changed

+812
-88
lines changed

7 files changed

+812
-88
lines changed

CMakeLists.txt

+2
Original file line numberDiff line numberDiff line change
@@ -503,6 +503,8 @@ endif()
503503
add_library(ggml OBJECT
504504
ggml.c
505505
ggml.h
506+
ggml-alloc.c
507+
ggml-alloc.h
506508
${GGML_SOURCES_CUDA}
507509
${GGML_SOURCES_OPENCL}
508510
${GGML_SOURCES_METAL}

Makefile

+6-1
Original file line numberDiff line numberDiff line change
@@ -329,7 +329,12 @@ $(info )
329329
ggml.o: ggml.c ggml.h ggml-cuda.h
330330
$(CC) $(CFLAGS) -c $< -o $@
331331

332-
llama.o: llama.cpp ggml.h ggml-cuda.h ggml-metal.h llama.h llama-util.h
332+
ggml-alloc.o: ggml-alloc.c ggml.h ggml-alloc.h
333+
$(CC) $(CFLAGS) -c $< -o $@
334+
335+
OBJS += ggml-alloc.o
336+
337+
llama.o: llama.cpp ggml.h ggml-alloc.h ggml-cuda.h ggml-metal.h llama.h llama-util.h
333338
$(CXX) $(CXXFLAGS) -c $< -o $@
334339

335340
common.o: examples/common.cpp examples/common.h

0 commit comments

Comments
 (0)