From 7ff83ea165f191161a94114b50a8b089afaa90c3 Mon Sep 17 00:00:00 2001 From: Andrew Woloszyn Date: Fri, 20 Dec 2024 10:00:18 -0800 Subject: [PATCH] [hip][cuda] Increase the size of the query pool. (#19542) Some of our models now have > 30k commands in them, and each command takes up 2 events, so increase the total so we don't run out. Signed-off-by: Andrew Woloszyn --- runtime/src/iree/hal/utils/stream_tracing.c | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/runtime/src/iree/hal/utils/stream_tracing.c b/runtime/src/iree/hal/utils/stream_tracing.c index f99e932cbb51..a60bdbb4ecd2 100644 --- a/runtime/src/iree/hal/utils/stream_tracing.c +++ b/runtime/src/iree/hal/utils/stream_tracing.c @@ -11,7 +11,7 @@ // Total number of events per tracing context. This translates to the maximum // number of outstanding timestamp queries before collection is required. // To prevent spilling pages we leave some room for the context structure. -#define IREE_HAL_TRACING_DEFAULT_QUERY_CAPACITY (32 * 1024 - 256) +#define IREE_HAL_TRACING_DEFAULT_QUERY_CAPACITY (128 * 1024 - 256) // iree_hal_stream_tracing_context_event_t contains a native event that is used // to record timestamps for tracing GPU execution. In this struct, there are