commit 178764110cded518ac92b76437c7983e8a2573b5
author Lei Zhang <antiagainst@google.com> Wed Mar 01 11:29:17 2023 -0800
committer Lei Zhang <antiagainst@google.com> Tue Jun 13 21:17:31 2023 -0700
tree 2fb38dc1f1aafb4333758afa432bc400f96bee2a
parent cc06c0b1206968f4cc4d338844d2e5dc6e752b8b

[metal] Add option to allow serial command dispatch for debugging

experimental/metal/registration/driver_module.c

diff --git a/experimental/metal/registration/driver_module.c b/experimental/metal/registration/driver_module.c
index 3126cfd..67a6595 100644
--- a/experimental/metal/registration/driver_module.c
+++ b/experimental/metal/registration/driver_module.c
@@ -11,9 +11,13 @@
 
 #include "experimental/metal/api.h"
 #include "iree/base/api.h"
+#include "iree/base/internal/flags.h"
 #include "iree/base/status.h"
 #include "iree/base/tracing.h"
 
+IREE_FLAG(bool, metal_serial_command_dispatch, false,
+          "Run all commands in command encoder sequentially");
+
 static iree_status_t iree_hal_metal_driver_factory_enumerate(
     void* self, iree_host_size_t* out_driver_info_count,
     const iree_hal_driver_info_t** out_driver_infos) {
@@ -47,6 +51,10 @@
 
   iree_hal_metal_device_params_t device_params;
   iree_hal_metal_device_params_initialize(&device_params);
+  device_params.command_dispatch_type =
+      FLAG_metal_serial_command_dispatch
+          ? IREE_HAL_METAL_COMMAND_DISPATCH_TYPE_SERIAL
+          : IREE_HAL_METAL_COMMAND_DISPATCH_TYPE_CONCURRENT;
 
   iree_status_t status = iree_hal_metal_driver_create(
       driver_name, &device_params, host_allocator, out_driver);