Add xformers-only estimate test

2026-08-13 13:15:52 -07:00 · 2026-08-13 13:15:52 -07:00 · b5b5c40fd7
parent 2bc2e8d73e
commit b5b5c40fd7
1 changed files with 5 additions and 0 deletions
--- a/tests-unit/comfy_test/memory_required_test.py
+++ b/tests-unit/comfy_test/memory_required_test.py
@ -43,6 +43,11 @@ def test_pytorch_flash_attention_uses_efficient_estimate(monkeypatch):
    assert _estimate() == EFFICIENT


+def test_xformers_uses_efficient_estimate(monkeypatch):
+    _patch_attention(monkeypatch, xformers=True)
+    assert _estimate() == EFFICIENT
+
+
 def test_flash_attention_flag_uses_efficient_estimate(monkeypatch):
    # --use-flash-attention must select the efficient estimate even when
    # pytorch attention was not auto enabled (e.g. torch builds without