fix: add missing sba capture when debug enabled

- In global bindless mode SBA may be programmed for scratch allocation,
missing sba capturing caused issues for kernels with scratch.
- this fix adds SBA capturing after SBA command

Related-To: NEO-7063

Signed-off-by: Mateusz Hoppe <mateusz.hoppe@intel.com>

shared/source/command_container/command_encoder_xehp_and_later.inl

@@ -323,6 +323,12 @@ void EncodeDispatchKernel<Family>::encode(CommandContainer &container, EncodeDis
             };
             EncodeStateBaseAddress<Family>::encode(encodeStateBaseAddressArgs);
             container.setDirtyStateForAllHeaps(false);
+
+            bool sbaTrackingEnabled = NEO::Debugger::isDebugEnabled(args.isInternal) && args.device->getL0Debugger();
+            NEO::EncodeStateBaseAddress<Family>::setSbaTrackingForL0DebuggerIfEnabled(sbaTrackingEnabled,
+                                                                                      *args.device,
+                                                                                      *container.getCommandStream(),
+                                                                                      sbaCmd, container.isUsingPrimaryBuffer());
         }
     }