From 6e1aeaf7abd1733409632571ddcc042a21a374bb Mon Sep 17 00:00:00 2001
From: dou <15529241576@163.com>
Date: Wed, 11 Sep 2024 11:30:24 +0800
Subject: [PATCH] fix: Error when running a non-exist op for Ascend NPU(#9303)

---
 ggml/src/ggml-cann.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ggml/src/ggml-cann.cpp b/ggml/src/ggml-cann.cpp
index 06930ba2e..24b8b752c 100644
--- a/ggml/src/ggml-cann.cpp
+++ b/ggml/src/ggml-cann.cpp
@@ -1942,7 +1942,7 @@ GGML_CALL ggml_backend_t ggml_backend_cann_init(int32_t device) {
         GGML_CANN_LOG_ERROR("%s: error: failed to allocate context\n", __func__);
         return nullptr;
     }
-
+    ggml_cann_set_device(ctx->device);
     ggml_backend_t cann_backend =
         new ggml_backend{/* .guid      = */ ggml_backend_cann_guid(),
                          /* .interface = */ ggml_backend_cann_interface,