diff options
Diffstat (limited to 'ggml/src/ggml-backend.c')
| -rw-r--r-- | ggml/src/ggml-backend.c | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/ggml/src/ggml-backend.c b/ggml/src/ggml-backend.c index e191c2d9..5903c89c 100644 --- a/ggml/src/ggml-backend.c +++ b/ggml/src/ggml-backend.c @@ -1,6 +1,7 @@ #include "ggml-backend-impl.h" #include "ggml-alloc.h" #include "ggml-impl.h" +#include "ggml-rpc.h" #include <assert.h> #include <limits.h> @@ -468,6 +469,10 @@ GGML_CALL static void ggml_backend_registry_init(void) { extern GGML_CALL int ggml_backend_cann_reg_devices(void); ggml_backend_cann_reg_devices(); #endif +#ifdef GGML_USE_RPC + extern GGML_CALL void ggml_backend_rpc_reg_devices(void); + ggml_backend_rpc_reg_devices(); +#endif } GGML_CALL void ggml_backend_register(const char * name, ggml_backend_init_fn init_fn, ggml_backend_buffer_type_t default_buffer_type, void * user_data) { @@ -943,6 +948,13 @@ GGML_CALL static ggml_backend_t ggml_backend_reg_cpu_init(const char * params, v GGML_UNUSED(user_data); } +GGML_CALL static ggml_backend_t ggml_backend_reg_rpc_init(const char* params, void* user_data) { + return ggml_backend_rpc_init((const char*)user_data); + + GGML_UNUSED(params); + GGML_UNUSED(user_data); +} + // multi-buffer buffer struct ggml_backend_multi_buffer_context { |