From e6f44577f469e891ed8dab1492a4c53224e0b765 Mon Sep 17 00:00:00 2001
From: Stefan Boberg <stefan.boberg@epicgames.com>
Date: Mon, 2 Dec 2024 12:21:53 +0100
Subject: added support for dynamic LLM tags (#245)

* added FLLMTag which can be used to register memory tags outside of core
* changed `UE_MEMSCOPE` -> `ZEN_MEMSCOPE` for consistency
* instrumented some subsystems with dynamic tags
---
 src/zenstore/cache/cachedisklayer.cpp | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

(limited to 'src/zenstore/cache/cachedisklayer.cpp')

diff --git a/src/zenstore/cache/cachedisklayer.cpp b/src/zenstore/cache/cachedisklayer.cpp
index 3046ab87c..a4f9fe78b 100644
--- a/src/zenstore/cache/cachedisklayer.cpp
+++ b/src/zenstore/cache/cachedisklayer.cpp
@@ -22,8 +22,18 @@
 
 //////////////////////////////////////////////////////////////////////////
 
+#include <zencore/memory/llm.h>
+
 namespace zen {
 
+const FLLMTag&
+GetCacheDiskTag()
+{
+	static FLLMTag _("disk", FLLMTag("cache"));
+
+	return _;
+}
+
 namespace {
 
 #pragma pack(push)
@@ -3894,6 +3904,8 @@ ZenCacheDiskLayer::DiscoverBuckets()
 	{
 		WorkLatch.AddCount(1);
 		Pool.ScheduleWork([this, &WorkLatch, &SyncLock, BucketPath]() {
+			ZEN_MEMSCOPE(GetCacheDiskTag());
+
 			auto			  _			 = MakeGuard([&]() { WorkLatch.CountDown(); });
 			const std::string BucketName = PathToUtf8(BucketPath.stem());
 			try
@@ -3980,6 +3992,7 @@ ZenCacheDiskLayer::Drop()
 void
 ZenCacheDiskLayer::Flush()
 {
+	ZEN_MEMSCOPE(GetCacheDiskTag());
 	ZEN_TRACE_CPU("Z$::Flush");
 
 	std::vector<CacheBucket*> Buckets;
@@ -4010,6 +4023,8 @@ ZenCacheDiskLayer::Flush()
 			{
 				WorkLatch.AddCount(1);
 				Pool.ScheduleWork([&WorkLatch, Bucket]() {
+					ZEN_MEMSCOPE(GetCacheDiskTag());
+
 					auto _ = MakeGuard([&]() { WorkLatch.CountDown(); });
 					try
 					{
-- 
cgit v1.2.3