Graphviper changes needed for astroviper demonstrator

[Jan-Willem]

No description provided.

[CLAassistant]

Thank you for your submission! We really appreciate it. Like many open source projects, we ask that you sign our Contributor License Agreement before we can accept your contribution.
_{You have signed the CLA already but the status is still pending? Let us recheck it.}

[codecov]

Codecov Report

❌ Patch coverage is 0% with 19 lines in your changes missing coverage. Please review.
✅ All tests successful. No failed tests found.

Files with missing lines	Patch %	Lines
src/toolviper/utils/memory_management.py	0.00%	19 Missing ⚠️

📢 Thoughts on this report? Let us know!

[Copilot]

Pull request overview

This PR adds a new toolviper.utils.memory_management module intended to support the “Graphviper / astroviper demonstrator” by providing small helpers for RSS reporting and returning memory to the OS via allocator-specific calls.

Changes:

• Added get_rss_gb() to report current process RSS via psutil.
• Added memory_setup() to tune glibc mallopt(M_MMAP_THRESHOLD, ...) on Linux.
• Added free_memory() to trigger GC and attempt allocator memory release (glibc malloc_trim on Linux; malloc_zone_pressure_relief on macOS).

Comments suppressed due to low confidence (2)

src/toolviper/utils/memory_management.py:40

• The macOS path calls malloc_zone_pressure_relief(None, 0) without defining argtypes/restype. With ctypes defaults, passing None is likely to raise a conversion TypeError (and the current broad except Exception: pass will silently swallow it), meaning this branch may never actually run. Define the function signature (void*/size_t) and pass a ctypes.c_void_p(0) NULL zone; also consider narrowing the exception handling and/or logging failures so unexpected issues aren't hidden.

            lib = ctypes.CDLL("libSystem.B.dylib")
            # malloc_zone_pressure_relief(zone=NULL, goal=0) asks all zones to
            # release as much memory as possible back to the OS.
            lib.malloc_zone_pressure_relief(None, 0)
        except Exception:
            pass

src/toolviper/utils/memory_management.py:39

• New memory-management utilities are introduced without tests. Since this repo has unit tests for other toolviper.utils modules, please add tests that validate (via monkeypatching sys.platform/ctypes) that memory_setup()/free_memory() are no-ops or safely handled on unsupported platforms, and that get_rss_gb() returns the expected unit conversion.

    On Linux this calls glibc's mallopt(M_MMAP_THRESHOLD, threshold).
    On macOS the system allocator handles this automatically; the call is skipped.
    """
    import sys, ctypes

    if sys.platform == "linux":
        ctypes.CDLL("libc.so.6").mallopt(-3, threshold)


def free_memory():
    """Return free memory pages to the OS.

    On Linux this calls glibc's malloc_trim(0).
    On macOS, malloc_zone_pressure_relief is used as the closest equivalent.
    """
    import sys, ctypes
    import gc

    gc.collect()
    if sys.platform == "linux":
        ctypes.CDLL("libc.so.6").malloc_trim(0)
    elif sys.platform == "darwin":
        try:
            lib = ctypes.CDLL("libSystem.B.dylib")
            # malloc_zone_pressure_relief(zone=NULL, goal=0) asks all zones to
            # release as much memory as possible back to the OS.
            lib.malloc_zone_pressure_relief(None, 0)
        except Exception:
            pass

---

💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.