fix it

Author: kumaraditya303

Python/tracemalloc.c

@@ -866,9 +866,6 @@ _PyTraceMalloc_Stop(void)
         goto done;
     }
 
-    /* stop tracing Python memory allocations */
-    _Py_atomic_store_int_relaxed(&tracemalloc_config.tracing, 0);
-
     /* unregister the hook on memory allocators */
     PyMem_SetAllocator(PYMEM_DOMAIN_RAW, &allocators.raw);
     PyMem_SetAllocator(PYMEM_DOMAIN_MEM, &allocators.mem);
@@ -882,6 +879,9 @@ _PyTraceMalloc_Stop(void)
 
     (void)PyRefTracer_SetTracer(NULL, NULL);
 
+    /* stop tracing Python memory allocations */
+    _Py_atomic_store_int_relaxed(&tracemalloc_config.tracing, 0);
+
 done:
     TABLES_UNLOCK();
 }
@@ -1213,6 +1213,13 @@ PyTraceMalloc_Track(unsigned int domain, uintptr_t ptr,
     }
     PyGILState_STATE gil_state = PyGILState_Ensure();
     TABLES_LOCK();
+    // Check again now that we hold the lock
+    if (!tracemalloc_config.tracing) {
+        TABLES_UNLOCK();
+        PyGILState_Release(gil_state);
+        /* tracemalloc is not tracing: do nothing */
+        return -2;
+    }
     int result = tracemalloc_add_trace_unlocked(domain, ptr, size);
     TABLES_UNLOCK();
     PyGILState_Release(gil_state);