commit 31c2d24c8960fb1296d126e25bba15cc4a5c4b30
author Lei Zhang <antiagainst@google.com> Mon Jun 12 19:48:48 2023 -0400
committer GitHub <noreply@github.com> Mon Jun 12 16:48:48 2023 -0700
tree 46e50d2cb1cfdf2780b3afe812a35d769fb3b663
parent c104444cb503f25852a5ea99b82c836ca4caa634

[cuda] Dump more synchronization related attributes (#14074)

Progress towards https://github.com/openxla/iree/issues/13245

experimental/cuda2/cuda_driver.c

diff --git a/experimental/cuda2/cuda_driver.c b/experimental/cuda2/cuda_driver.c
index d06d6ab..40c09a4 100644
--- a/experimental/cuda2/cuda_driver.c
+++ b/experimental/cuda2/cuda_driver.c
@@ -387,6 +387,25 @@
   IREE_RETURN_IF_ERROR(iree_string_builder_append_format(
       builder, "\n- memory-l2-cache-size: %d bytes", l2_cache_size));
 
+  int supports_64bit_memops = 0;
+  IREE_CUDA_QUERY_ATTRIBUTE(CAN_USE_64_BIT_STREAM_MEM_OPS,
+                            supports_64bit_memops);
+  int supports_timeline_semaphore_interop = 0;
+  IREE_CUDA_QUERY_ATTRIBUTE(TIMELINE_SEMAPHORE_INTEROP_SUPPORTED,
+                            supports_timeline_semaphore_interop);
+  int mem_sync_domain_count = 0;
+  IREE_CUDA_QUERY_ATTRIBUTE(MEM_SYNC_DOMAIN_COUNT, mem_sync_domain_count);
+
+  IREE_RETURN_IF_ERROR(iree_string_builder_append_cstring(builder, "\n"));
+  IREE_RETURN_IF_ERROR(iree_string_builder_append_format(
+      builder, "\n- sync-supports-64-bit-stream-mem-ops: %d",
+      supports_64bit_memops));
+  IREE_RETURN_IF_ERROR(iree_string_builder_append_format(
+      builder, "\n- sync-supports-timeline-semaphore-interop: %d",
+      supports_timeline_semaphore_interop));
+  IREE_RETURN_IF_ERROR(iree_string_builder_append_format(
+      builder, "\n- sync-mem-domain-count: %d", mem_sync_domain_count));
+
   // Other GPU characteristics.
   int multiprocessor_count = 0;
   IREE_CUDA_QUERY_ATTRIBUTE(MULTIPROCESSOR_COUNT, multiprocessor_count);