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

bachelor-dou · web-flow · commit 904837e0cb2f · 2024-09-25T11:30:38.000+08:00
diff --git 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));
         }

Original file line number	Diff line number	Diff line change
`@@ -227,6 +227,7 @@ struct ggml_backend_cann_context {`
`227`	`227`	`* @brief Destructor for cleaning up resources.`
`228`	`228`	`*/`
`229`	`229`	`~ggml_backend_cann_context() {`
	`230`	`+ ggml_cann_set_device(device);`
`230`	`231`	`if (copy_event != nullptr) {`
`231`	`232`	`ACL_CHECK(aclrtDestroyEvent(copy_event));`
`232`	`233`	`}`