rubra-ai
diff --git a/‎ggml/include/ggml-cann.h
+2 b/‎ggml/include/ggml-cann.h
+2
diff --git a/‎ggml/src/ggml-backend.cpp
+8-1 b/‎ggml/src/ggml-backend.cpp
+8-1
@@ -34,6 +34,8 @@ extern "C" {
  */
 #define GGML_CANN_MAX_DEVICES 16
 
+GGML_API ggml_backend_reg_t ggml_backend_cann_reg(void);
+
 /**
  * @brief Initializes the CANN backend for a specified device.
  *
 
@@ -561,6 +561,10 @@ void * ggml_backend_reg_get_proc_address(ggml_backend_reg_t reg, const char * na
 #  include "ggml-amx.h"
 #endif
 
+#ifdef GGML_USE_CANN
+#include "ggml-cann.h"
+#endif
+
 struct ggml_backend_registry {
     std::vector<ggml_backend_reg_t> backends;
     std::vector<ggml_backend_dev_t> devices;
@@ -587,8 +591,11 @@ struct ggml_backend_registry {
 #ifdef GGML_USE_AMX
         register_backend(ggml_backend_amx_reg());
 #endif
+#ifdef GGML_USE_CANN
+        register_backend(ggml_backend_cann_reg());
+#endif
 
-        // TODO: kompute, cann
+        // TODO: kompute
 
         register_backend(ggml_backend_cpu_reg());
     }
Original file line number	Diff line number	Diff line change
`@@ -34,6 +34,8 @@ extern "C" {`
`34`	`34`	`*/`
`35`	`35`	`#define GGML_CANN_MAX_DEVICES 16`
`36`	`36`
	`37`	`+GGML_API ggml_backend_reg_t ggml_backend_cann_reg(void);`
	`38`	`+`
`37`	`39`	`/**`
`38`	`40`	`* @brief Initializes the CANN backend for a specified device.`
`39`	`41`	`*`