cann: fix crash when llama-bench is running on multiple cann devices (#9627)
This commit is contained in:
parent
70392f1f81
commit
904837e0cb
1 changed files with 1 additions and 0 deletions
|
@ -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));
|
||||
}
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue