diff --git a/src/intel/ds/intel_tracepoints.py b/src/intel/ds/intel_tracepoints.py
index 6ce8dec611d..73a3bc7bffa 100644
--- a/src/intel/ds/intel_tracepoints.py
+++ b/src/intel/ds/intel_tracepoints.py
@@ -128,6 +128,7 @@ def define_tracepoints(args):
 
     # STATE_BASE_ADDRESS emission, only for Anv
     begin_end_tp('sba',
+                 tp_args=[Arg(type='uint8_t', var='mode', c_format='%hhu'),],
                  end_pipelined=False)
 
     # Dynamic rendering tracepoints, only for Anv
diff --git a/src/intel/vulkan/genX_cmd_buffer.c b/src/intel/vulkan/genX_cmd_buffer.c
index f827ba57ce8..77f22449c92 100644
--- a/src/intel/vulkan/genX_cmd_buffer.c
+++ b/src/intel/vulkan/genX_cmd_buffer.c
@@ -381,7 +381,8 @@ genX(cmd_buffer_emit_state_base_address)(struct anv_cmd_buffer *cmd_buffer)
 
    cmd_buffer->batch.trace = tmp_trace;
 
-   trace_intel_end_sba(cmd_buffer->batch.trace);
+   trace_intel_end_sba(cmd_buffer->batch.trace,
+                       cmd_buffer->state.pending_db_mode);
 #endif
 
 #if GFX_VERx10 >= 125