eagerly free large objects

cortex-engine · Sep 12, 2024 · 9ecfa08 · 9ecfa08
1 parent 1b53779
commit 9ecfa08
Show file tree

Hide file tree

Showing 4 changed files with 23 additions and 9 deletions.
diff --git a/include/hx/GC.h b/include/hx/GC.h
@@ -24,6 +24,7 @@
 #ifdef HXCPP_TELEMETRY
 extern void __hxt_gc_new(hx::StackContext *inStack, void* obj, int inSize, const char *inName);
 extern void __hxt_gc_alloc(void* obj, int inSize);
+extern void __hxt_gc_free_large(void* obj);
 #endif
 
 

diff --git a/src/hx/Telemetry.cpp b/src/hx/Telemetry.cpp
@@ -650,6 +650,7 @@ void __hxt_new_hash(void* obj, int inSize)
   #endif
 }
 void __hxt_gc_alloc(void* obj, int inSize) { }
+void __hxt_gc_free_large(void*) {}
 void __hxt_gc_new(hx::StackContext *stack, void* obj, int inSize, const char* name)
 {
   #ifdef HXCPP_STACK_TRACE

diff --git a/src/hx/TelemetryTracy.cpp b/src/hx/TelemetryTracy.cpp
@@ -82,15 +82,6 @@ void __hxt_gc_alloc(void* obj, int inSize)
 		{
 			AutoLock lock(largeAllocsLock);
 
-			// Skip multiple large object allocations since they can be recycled in-between GC collections.
-			for (auto i = 0; i < largeAllocs.size(); i++)
-			{
-				if (largeAllocs[i] == obj)
-				{
-					return;
-				}
-			}
-
 			largeAllocs.push(obj);
 
 			TracyAllocN(obj, inSize, lohName);
@@ -104,6 +95,23 @@ void __hxt_gc_alloc(void* obj, int inSize)
 	#endif
 }
 
+void __hxt_gc_free_large(void* obj)
+{
+	AutoLock lock(largeAllocsLock);
+
+	for (auto i = 0; i < largeAllocs.size(); i++)
+	{
+		if (largeAllocs[i] == obj)
+		{
+			largeAllocs.erase(i);
+
+			TracyFreeN(obj, lohName);
+
+			return;
+		}
+	}
+}
+
 void __hxt_gc_realloc(void* oldObj, void* newObj, int newSize) { }
 
 void __hxt_gc_after_mark(int byteMarkId, int endianByteMarkId)

diff --git a/src/hx/gc/Immix.cpp b/src/hx/gc/Immix.cpp
@@ -3179,6 +3179,10 @@ class GlobalAllocator
 
    void FreeLarge(void *inLarge)
    {
+#ifdef HXCPP_TELEMETRY
+       __hxt_gc_free_large(inLarge);
+#endif
+
       ((unsigned char *)inLarge)[HX_ENDIAN_MARK_ID_BYTE] = 0;
       // AllocLarge will not lock this list unless it decides there is a suitable
       //  value, so we can't doa realloc without potentially crashing it.