remove dependence of ggml-opt.cpp on ggml-cpu.h

ggerganov · Nov 16, 2024 · 4685431 · 4685431
1 parent 25b21f6
commit 4685431
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/src/ggml-opt.cpp b/src/ggml-opt.cpp
@@ -3,7 +3,6 @@
 #include "ggml.h"
 #include "ggml-alloc.h"
 #include "ggml-backend.h"
-#include "ggml-cpu.h"
 #include "ggml-impl.h"
 
 #include <algorithm>
@@ -105,9 +104,7 @@ ggml_opt_dataset_t ggml_opt_dataset_init(int64_t ne_datapoint, int64_t ne_label,
         result->nbs_labels = 0;
     }
 
-    result->backend = ggml_backend_cpu_init();
-    GGML_ASSERT(result->backend);
-    result->buf = ggml_backend_alloc_ctx_tensors(result->ctx, result->backend);
+    result->buf = ggml_backend_alloc_ctx_tensors_from_buft(result->ctx, ggml_backend_cpu_buffer_type());
 
     const int64_t nshards = ndata/ndata_shard;
     result->permutation.resize(nshards);