commit 40876b9e72639b7369a60b42fe6a22db776465f0
author Ben Vanik <ben.vanik@gmail.com> Mon Mar 06 11:26:53 2023 -0800
committer Ben Vanik <ben.vanik@gmail.com> Tue Mar 07 09:37:59 2023 -0800
tree f50a9ab8378462320fcd7f8734d0e5525b3279b7
parent fd76fb45d8be2445ed1d39bb5c2b533eb9838626

Write out element strings instead of integers in python tracing.

runtime/bindings/python/vm.cc

diff --git a/runtime/bindings/python/vm.cc b/runtime/bindings/python/vm.cc
index 3e32df9..41e979b 100644
--- a/runtime/bindings/python/vm.cc
+++ b/runtime/bindings/python/vm.cc
@@ -367,8 +367,14 @@
       // Element type.
       iree_hal_element_type_t element_type =
           iree_hal_buffer_view_element_type(buffer_view);
-      // TODO: Would be nice to output as hex.
-      record["element_type"] = element_type;
+      char element_type_str[64] = {0};
+      iree_host_size_t element_type_length = 0;
+      CheckApiStatus(
+          iree_hal_format_element_type(element_type, sizeof(element_type_str),
+                                       element_type_str, &element_type_length),
+          "Formatting element type");
+      record["element_type"] =
+          std::string(element_type_str, element_type_length);
 
       // Map memory.
       iree_device_size_t byte_length = iree_hal_buffer_byte_length(raw_buffer);