cann: fix crash when llama-bench is running on multiple cann devices (#9627)

2024-09-25 11:30:38 +08:00 · 2024-09-25 11:30:38 +08:00 · 904837e0cb
commit 904837e0cb
parent 70392f1f81
1 changed files with 1 additions and 0 deletions
--- a/ggml/src/ggml-cann/common.h
+++ b/ggml/src/ggml-cann/common.h
@ -227,6 +227,7 @@ struct ggml_backend_cann_context {
     * @brief Destructor for cleaning up resources.
     */
    ~ggml_backend_cann_context() {
+        ggml_cann_set_device(device);
        if (copy_event != nullptr) {
            ACL_CHECK(aclrtDestroyEvent(copy_event));
        }