ggml-zdnn: fix invalid ztensor buffer release

Signed-off-by: Aaron Teo <aaron.teo1@ibm.com>
This commit is contained in:
Aaron Teo 2025-07-29 00:09:00 +08:00
parent 2cfa118fa9
commit 2872276d8a
No known key found for this signature in database
1 changed files with 3 additions and 0 deletions

View File

@ -321,6 +321,9 @@ static void ggml_backend_zdnn_buffer_free_buffer(ggml_backend_buffer_t buffer) {
ggml_backend_zdnn_buffer_context * ctx = (ggml_backend_zdnn_buffer_context *)buffer->context;
for (int i = 0; i < ctx->n_buffers; i++) {
if (&ctx->buffers[i].ztensor.buffer == NULL) {
continue;
}
ZDNN_CHECK(zdnn_free_ztensor_buffer(&ctx->buffers[i].ztensor));
}