]> git.djapps.eu Git - pkg/ggml/sources/ggml/commitdiff
cont : fixes
authorGeorgi Gerganov <redacted>
Thu, 3 Oct 2024 19:03:05 +0000 (22:03 +0300)
committerGeorgi Gerganov <redacted>
Thu, 3 Oct 2024 19:12:49 +0000 (22:12 +0300)
examples/mnist/mnist-common.h
tests/test-backend-buffer.cpp

index 0a88f05d391a8d17903e18b48f00218c8ee6cfa2..37531b2857d36450ed9c38cfc18e12b1f6fa6348 100644 (file)
@@ -57,8 +57,8 @@ struct mnist_model {
     ggml_backend_buffer_t buf_compute = nullptr;
 
     mnist_model(const std::string & backend_name) {
-        const ggml_backend_reg_t reg = ggml_backend_reg_by_name(backend_name.c_str());
-        if (reg == nullptr) {
+        const ggml_backend_dev_t dev = ggml_backend_dev_by_name(backend_name.c_str());
+        if (dev == nullptr) {
             fprintf(stderr, "%s: ERROR: backend %s not found, available:\n", __func__, backend_name.c_str());
             for (size_t i = 0; i < ggml_backend_reg_count(); ++i) {
                 fprintf(stderr, "  - %s\n", ggml_backend_reg_name(ggml_backend_reg_get(i)));
@@ -68,7 +68,7 @@ struct mnist_model {
 
         fprintf(stderr, "%s: using %s backend\n", __func__, backend_name.c_str());
 
-        ggml_backend_t backend = ggml_backend_init_by_name(backend_name.c_str(), NULL);
+        ggml_backend_t backend = ggml_backend_dev_init(dev, NULL);
         if (ggml_backend_is_cpu(backend)) {
             const int ncores_logical = std::thread::hardware_concurrency();
             ggml_backend_cpu_set_n_threads(backend, std::min(ncores_logical, (ncores_logical + 4)/2));
index cd53358cec59830b63ca73dcc2cfb35ab8d457b2..09f46e8b3f411a0bf69ae54a77cb0ea283877c39 100644 (file)
@@ -67,10 +67,12 @@ int main() {
     printf("Testing %zu backends\n\n", ggml_backend_reg_count());
 
     for (size_t i = 0; i < ggml_backend_reg_count(); i++) {
-        const char * name = ggml_backend_reg_name(ggml_backend_reg_get(i));
-        printf("Backend %zu/%zu (%s)\n", i + 1, ggml_backend_reg_count(), name);
+        ggml_backend_reg_t reg = ggml_backend_reg_get(i);
+        const char * reg_name = ggml_backend_reg_name(reg);
+        printf("Backend %zu/%zu (%s)\n", i + 1, ggml_backend_reg_count(), reg_name);
 
-        ggml_backend_t backend = ggml_backend_init_by_name(name, NULL);
+        const ggml_backend_dev_t dev = ggml_backend_reg_dev_get(reg, 0);
+        ggml_backend_t backend = ggml_backend_dev_init(dev, NULL);
         GGML_ASSERT(backend != NULL);
         printf("  Backend name: %s\n", ggml_backend_name(backend));