Преглед изворни кода

cann: Fix error when running a non-exist op (#9424)

Xinpeng Dou пре 1 година
родитељ
комит
df4b7945ae
1 измењених фајлова са 1 додато и 1 уклоњено
  1. 1 1
      ggml/src/ggml-cann.cpp

+ 1 - 1
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__);
         GGML_CANN_LOG_ERROR("%s: error: failed to allocate context\n", __func__);
         return nullptr;
         return nullptr;
     }
     }
-
+    ggml_cann_set_device(ctx->device);
     ggml_backend_t cann_backend =
     ggml_backend_t cann_backend =
         new ggml_backend{/* .guid      = */ ggml_backend_cann_guid(),
         new ggml_backend{/* .guid      = */ ggml_backend_cann_guid(),
                          /* .interface = */ ggml_backend_cann_interface,
                          /* .interface = */ ggml_backend_cann_interface,