fix: symmetric compatibility (#555)

gsprochette · web-flow · commit 3623d08f0a09 · 2026-03-02T18:03:33.000+01:00
diff --git a/src/pruna/algorithms/deepcache.py b/src/pruna/algorithms/deepcache.py
@@ -50,8 +50,21 @@ class DeepCache(PrunaAlgorithmBase):
         "diffusers_int8",
         "quanto",
         "sage_attn",
+        "hyper",
+        "padding_pruning",
         ]
-    compatible_after: Iterable[str] = ["stable_fast", "torch_compile"]
+    compatible_after: Iterable[str] = [
+        "stable_fast",
+        "torch_compile",
+        "img2img_denoise",
+        "realesrgan_upscale",
+        "text_to_image_distillation_inplace_perp",
+        "text_to_image_distillation_lora",
+        "text_to_image_distillation_perp",
+        "text_to_image_inplace_perp",
+        "text_to_image_lora",
+        "text_to_image_perp",
+    ]
 
     def get_hyperparameters(self) -> list:
         """
diff --git a/src/pruna/algorithms/denoise.py b/src/pruna/algorithms/denoise.py
@@ -75,6 +75,7 @@ class Img2ImgDenoise(PrunaAlgorithmBase):
         "torchao",
         "qkv_diffusers",
         "ring_attn",
+        "hyper",
     ]
 
     def get_hyperparameters(self) -> list:
diff --git a/src/pruna/algorithms/fastercache.py b/src/pruna/algorithms/fastercache.py
@@ -57,7 +57,8 @@ class FasterCache(PrunaAlgorithmBase):
     processor_required: bool = False
     dataset_required: bool = False
     runs_on: list[str] = ["cpu", "cuda", "accelerate"]
-    compatible_before: Iterable[str] = ["hqq_diffusers", "diffusers_int8", "sage_attn"]
+    compatible_before: Iterable[str] = ["hqq_diffusers", "diffusers_int8", "sage_attn", "hyper", "padding_pruning"]
+    compatible_after: Iterable[str] = ["img2img_denoise", "realesrgan_upscale"]
 
     def get_hyperparameters(self) -> list:
         """
diff --git a/src/pruna/algorithms/flash_attn3.py b/src/pruna/algorithms/flash_attn3.py
@@ -51,7 +51,7 @@ class FlashAttn3(PrunaAlgorithmBase):
     processor_required: bool = False
     runs_on: list[str] = ["cuda", "accelerate"]
     dataset_required: bool = False
-    compatible_before: Iterable[str] = ["torchao"]
+    compatible_before: Iterable[str] = ["torchao", "padding_pruning"]
     compatible_after: Iterable[str] = ["fora", "torch_compile"]
 
     def model_check_fn(self, model: Any) -> bool:
diff --git a/src/pruna/algorithms/fora.py b/src/pruna/algorithms/fora.py
@@ -50,9 +50,11 @@ class FORA(PrunaAlgorithmBase):
         "hqq_diffusers",
         "torchao",
         "flash_attn3",
-        "sage_attn"
+        "sage_attn",
+        "hyper",
+        "padding_pruning",
     ]
-    compatible_after: Iterable[str] = ["stable_fast", "torch_compile"]
+    compatible_after: Iterable[str] = ["stable_fast", "torch_compile", "img2img_denoise", "realesrgan_upscale"]
 
     def get_hyperparameters(self) -> list:
         """
@@ -80,9 +82,7 @@ def get_hyperparameters(self) -> list:
                 "backbone_calls_per_step",
                 sequence=range(1, 4),
                 default_value=1,
-                meta=dict(
-                    desc="Number of backbone forward passes per diffusion step (e.g., 2 for CFG)."
-                ),
+                meta=dict(desc="Number of backbone forward passes per diffusion step (e.g., 2 for CFG)."),
             ),
         ]
 
diff --git a/src/pruna/algorithms/half.py b/src/pruna/algorithms/half.py
@@ -52,6 +52,12 @@ class Half(PrunaAlgorithmBase):
         "ifw",
         "whisper_s2t",
         "sage_attn",
+        "hyper",
+        "ipex_llm",
+        "text_to_text_inplace_perp",
+        "text_to_text_lora",
+        "text_to_text_perp",
+        "x_fast",
     ]
 
     def model_check_fn(self, model: Any) -> bool:
diff --git a/src/pruna/algorithms/hqq_diffusers.py b/src/pruna/algorithms/hqq_diffusers.py
@@ -64,8 +64,17 @@ class HQQDiffusers(PrunaAlgorithmBase):
     processor_required: bool = False
     runs_on: list[str] = ["cuda"]
     dataset_required: bool = False
-    compatible_before: Iterable[str] = ["qkv_diffusers"]
-    compatible_after: Iterable[str] = ["deepcache", "fastercache", "fora", "pab", "torch_compile", "sage_attn"]
+    compatible_before: Iterable[str] = ["qkv_diffusers", "padding_pruning"]
+    compatible_after: Iterable[str] = [
+        "deepcache",
+        "fastercache",
+        "fora",
+        "pab",
+        "torch_compile",
+        "sage_attn",
+        "img2img_denoise",
+        "realesrgan_upscale",
+    ]
     disjointly_compatible_before: Iterable[str] = []
     disjointly_compatible_after: Iterable[str] = ["torchao"]
 
diff --git a/src/pruna/algorithms/huggingface_diffusers_int8.py b/src/pruna/algorithms/huggingface_diffusers_int8.py
@@ -60,8 +60,18 @@ class DiffusersInt8(PrunaAlgorithmBase):
     dataset_required: bool = False
     runs_on: list[str] = ["cuda", "accelerate"]
     save_fn: None = None
-    compatible_before: Iterable[str] = ["qkv_diffusers"]
-    compatible_after: Iterable[str] = ["deepcache", "fastercache", "fora", "pab", "torch_compile", "sage_attn"]
+    compatible_before: Iterable[str] = ["qkv_diffusers", "padding_pruning"]
+    compatible_after: Iterable[str] = [
+        "deepcache",
+        "fastercache",
+        "fora",
+        "pab",
+        "torch_compile",
+        "sage_attn",
+        "img2img_denoise",
+        "hyper",
+        "realesrgan_upscale",
+    ]
 
     def get_hyperparameters(self) -> list:
         """
diff --git a/src/pruna/algorithms/llm_compressor.py b/src/pruna/algorithms/llm_compressor.py
@@ -52,7 +52,7 @@ class LLMCompressor(PrunaAlgorithmBase):
     dataset_required: bool = True
     runs_on: list[str] = ["cuda"]
     compatible_before: Iterable[str] = []
-    compatible_after: Iterable[str] = []
+    compatible_after: Iterable[str] = ["sage_attn"]
 
     def get_hyperparameters(self) -> list:
         """
diff --git a/src/pruna/algorithms/pab.py b/src/pruna/algorithms/pab.py
@@ -53,8 +53,8 @@ class PAB(PrunaAlgorithmBase):
     processor_required: bool = False
     dataset_required: bool = False
     runs_on: list[str] = ["cpu", "cuda", "accelerate"]
-    compatible_before: Iterable[str] = ["hqq_diffusers", "diffusers_int8", "sage_attn"]
-    compatible_after: Iterable[str] = []
+    compatible_before: Iterable[str] = ["hqq_diffusers", "diffusers_int8", "sage_attn", "hyper", "padding_pruning"]
+    compatible_after: Iterable[str] = ["img2img_denoise", "realesrgan_upscale"]
 
     def get_hyperparameters(self) -> list:
         """
diff --git a/src/pruna/algorithms/qkv_diffusers.py b/src/pruna/algorithms/qkv_diffusers.py
@@ -51,6 +51,9 @@ class QKVFusing(PrunaAlgorithmBase):
         "fora",
         "torch_compile",
         "ring_attn",
+        "img2img_denoise",
+        "padding_pruning",
+        "realesrgan_upscale",
     ]
 
     def model_check_fn(self, model: Any) -> bool:
diff --git a/src/pruna/algorithms/quanto.py b/src/pruna/algorithms/quanto.py
@@ -50,7 +50,20 @@ class Quanto(PrunaAlgorithmBase):
     dataset_required: bool = False
     runs_on: list[str] = ["cuda"]
     compatible_before: Iterable[str] = ["qkv_diffusers"]
-    compatible_after: Iterable[str] = ["deepcache", "sage_attn"]
+    compatible_after: Iterable[str] = [
+        "deepcache",
+        "sage_attn",
+        "text_to_image_distillation_inplace_perp",
+        "text_to_image_distillation_lora",
+        "text_to_image_distillation_perp",
+        "text_to_image_inplace_perp",
+        "text_to_image_lora",
+        "text_to_image_perp",
+        "text_to_text_inplace_perp",
+        "text_to_text_lora",
+        "text_to_text_perp",
+        "x_fast",
+    ]
 
     def get_hyperparameters(self) -> list:
         """
diff --git a/src/pruna/algorithms/ring_attn/ring.py b/src/pruna/algorithms/ring_attn/ring.py
@@ -64,7 +64,7 @@ class RingAttn(PrunaAlgorithmBase):
         "qkv_diffusers",
         "padding_pruning",
     ]
-    compatible_after: Iterable[str | AlgorithmTag] = ["torch_compile"]
+    compatible_after: Iterable[str | AlgorithmTag] = ["torch_compile", "img2img_denoise", "realesrgan_upscale"]
 
     def get_hyperparameters(self) -> list:
         """
diff --git a/src/pruna/algorithms/stable_fast.py b/src/pruna/algorithms/stable_fast.py
@@ -40,7 +40,8 @@ class StableFast(PrunaAlgorithmBase):
     processor_required: bool = False
     runs_on: list[str] = ["cuda"]
     dataset_required: bool = False
-    compatible_before: Iterable[str] = ["half", "deepcache", "fora"]
+    compatible_before: Iterable[str] = ["half", "deepcache", "fora", "hyper", "padding_pruning"]
+    compatible_after: Iterable[str] = ["img2img_denoise", "realesrgan_upscale"]
     required_install: str = "``pip install pruna[stable-fast]``"
 
     def model_check_fn(self, model: Any) -> bool:
diff --git a/src/pruna/algorithms/torch_compile/torch_compile.py b/src/pruna/algorithms/torch_compile/torch_compile.py
@@ -71,7 +71,20 @@ class TorchCompile(PrunaAlgorithmBase):
         "deepcache",
         "fora",
         "sage_attn",
+        "hyper",
+        "padding_pruning",
+        "ring_attn",
+        "text_to_image_distillation_inplace_perp",
+        "text_to_image_distillation_lora",
+        "text_to_image_distillation_perp",
+        "text_to_image_inplace_perp",
+        "text_to_image_lora",
+        "text_to_image_perp",
+        "text_to_text_inplace_perp",
+        "text_to_text_lora",
+        "text_to_text_perp",
     ]
+    compatible_after: Iterable[str] = ["img2img_denoise", "realesrgan_upscale"]
 
     def get_hyperparameters(self) -> list:
         """
diff --git a/src/pruna/algorithms/torch_dynamic.py b/src/pruna/algorithms/torch_dynamic.py
@@ -43,7 +43,18 @@ class TorchDynamic(PrunaAlgorithmBase):
     runs_on: list[str] = ["cpu", "cuda"]
     dataset_required: bool = False
     compatible_before: Iterable[str] = []
-    compatible_after: Iterable[str] = ["sage_attn"]
+    compatible_after: Iterable[str] = [
+        "sage_attn",
+        "text_to_image_distillation_inplace_perp",
+        "text_to_image_distillation_lora",
+        "text_to_image_distillation_perp",
+        "text_to_image_inplace_perp",
+        "text_to_image_lora",
+        "text_to_image_perp",
+        "text_to_text_inplace_perp",
+        "text_to_text_lora",
+        "text_to_text_perp",
+    ]
 
     def get_hyperparameters(self) -> list:
         """
diff --git a/src/pruna/algorithms/torchao.py b/src/pruna/algorithms/torchao.py
@@ -93,8 +93,15 @@ class Torchao(PrunaAlgorithmBase):
     processor_required: bool = False
     runs_on: list[str] = ["cpu", "cuda", "accelerate"]
     dataset_required: bool = False
-    compatible_before: Iterable[str] = ["qkv_diffusers", "torch_structured"]
-    compatible_after: Iterable[str] = ["flash_attn3", "fora", "torch_compile", "sage_attn"]
+    compatible_before: Iterable[str] = ["qkv_diffusers", "torch_structured", "padding_pruning"]
+    compatible_after: Iterable[str] = [
+        "flash_attn3",
+        "fora",
+        "torch_compile",
+        "sage_attn",
+        "img2img_denoise",
+        "realesrgan_upscale",
+    ]
     disjointly_compatible_before: Iterable[str] = ["hqq", "hqq_diffusers"]
     disjointly_compatible_after: Iterable[str] = []
 
diff --git a/src/pruna/algorithms/upscale.py b/src/pruna/algorithms/upscale.py
@@ -77,7 +77,8 @@ class RealESRGAN(PrunaAlgorithmBase):
         "diffusers_int8",
         "torchao",
         "qkv_diffusers",
-        "ring_attn"
+        "ring_attn",
+        "hyper",
     ]
 
     def get_hyperparameters(self) -> list:

Original file line number	Diff line number	Diff line change
`@@ -75,6 +75,7 @@ class Img2ImgDenoise(PrunaAlgorithmBase):`
`75`	`75`	`"torchao",`
`76`	`76`	`"qkv_diffusers",`
`77`	`77`	`"ring_attn",`
	`78`	`+ "hyper",`
`78`	`79`	`]`
`79`	`80`
`80`	`81`	`def get_hyperparameters(self) -> list:`