commit 830820e4c437e94cf62a1b477db86b5f7abc4c30
author Ben Vanik <ben.vanik@gmail.com> Tue Dec 20 10:44:06 2022 -0800
committer GitHub <noreply@github.com> Tue Dec 20 18:44:06 2022 +0000
tree d5d69f162c8352547279673ea456c57410981c85
parent d985a1b8faf7242a61dd1e289bbddd8e3780c3d0

Adding tracing annotations for VMA buffer allocation sizes. (#11621)

runtime/src/iree/hal/allocator.c

diff --git a/runtime/src/iree/hal/allocator.c b/runtime/src/iree/hal/allocator.c
index daa418b..59ecffb 100644
--- a/runtime/src/iree/hal/allocator.c
+++ b/runtime/src/iree/hal/allocator.c
@@ -124,6 +124,7 @@
   IREE_ASSERT_ARGUMENT(out_buffer);
   *out_buffer = NULL;
   IREE_TRACE_ZONE_BEGIN(z0);
+  IREE_TRACE_ZONE_APPEND_VALUE(z0, (int64_t)allocation_size);
   iree_hal_buffer_params_canonicalize(&params);
   iree_status_t status = _VTABLE_DISPATCH(allocator, allocate_buffer)(
       allocator, &params, allocation_size, initial_data, out_buffer);

runtime/src/iree/hal/drivers/vulkan/vma_buffer.cc

diff --git a/runtime/src/iree/hal/drivers/vulkan/vma_buffer.cc b/runtime/src/iree/hal/drivers/vulkan/vma_buffer.cc
index 4ff42fa..f119837 100644
--- a/runtime/src/iree/hal/drivers/vulkan/vma_buffer.cc
+++ b/runtime/src/iree/hal/drivers/vulkan/vma_buffer.cc
@@ -46,6 +46,7 @@
   IREE_ASSERT_ARGUMENT(allocation);
   IREE_ASSERT_ARGUMENT(out_buffer);
   IREE_TRACE_ZONE_BEGIN(z0);
+  IREE_TRACE_ZONE_APPEND_VALUE(z0, (int64_t)allocation_size);
 
   iree_allocator_t host_allocator =
       iree_hal_allocator_host_allocator(allocator);
@@ -84,6 +85,8 @@
       iree_hal_vulkan_vma_buffer_cast(base_buffer);
   iree_allocator_t host_allocator = base_buffer->host_allocator;
   IREE_TRACE_ZONE_BEGIN(z0);
+  IREE_TRACE_ZONE_APPEND_VALUE(
+      z0, (int64_t)iree_hal_buffer_allocation_size(base_buffer));
 
   // IREE_TRACE_FREE_NAMED("VMA", (void*)buffer->handle);