diff --git a/warp/_src/context.py b/warp/_src/context.py index 97afe7cd2c..15159846ee 100644 --- a/warp/_src/context.py +++ b/warp/_src/context.py @@ -3858,6 +3858,12 @@ def __init__(self, runtime, alias, ordinal=-1, is_primary=False, context=None): if warp.config.enable_mempools_at_init: # enable if supported self.is_mempool_enabled = self.is_mempool_supported + elif self.is_hip and self.is_mempool_supported: + # HIP/ROCm: enable mempool by default when graph capture is supported + # (hipGraph requires mempool for in-capture allocations). + # Previously disabled due to hipMemsetAsync unreliability, but + # graph capture is now validated on ROCm 7.2 (Warp PR #15). + self.is_mempool_enabled = True else: # disable by default self.is_mempool_enabled = False