From cc2c883aa136b2bb3bcb78b77ad6811da44ee98c Mon Sep 17 00:00:00 2001 From: ERMAN GURSES Date: Wed, 14 Aug 2024 07:31:45 -0700 Subject: [PATCH] Add deinit for expeted and actual results Signed-off-by: ERMAN GURSES --- tools/testing/e2e/iree-e2e-attention-test.cc | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tools/testing/e2e/iree-e2e-attention-test.cc b/tools/testing/e2e/iree-e2e-attention-test.cc index 4993497e8e714..7f9ecc9c09b74 100644 --- a/tools/testing/e2e/iree-e2e-attention-test.cc +++ b/tools/testing/e2e/iree-e2e-attention-test.cc @@ -8,7 +8,6 @@ #include #include #include -#include #include "iree/base/api.h" #include "iree/base/internal/cpu.h" @@ -278,6 +277,9 @@ static void attention_results_deinitialize(attention_results_t* results) { iree_allocator_free(results->host_allocator, results->query_contents.data); iree_allocator_free(results->host_allocator, results->key_contents.data); iree_allocator_free(results->host_allocator, results->value_contents.data); + iree_allocator_free(results->host_allocator, results->actual_contents.data); + iree_allocator_free(results->host_allocator, results->expected_contents.data); + IREE_TRACE_ZONE_END(z0); }