add xpu to valid hardware for torch.compile (#42079)

sywangyi · web-flow · commit d012f34e0ddb · 2025-11-12T10:12:05.000+01:00
Signed-off-by: Wang, Yi A &lt;yi.a.wang@intel.com&gt;
diff --git a/src/transformers/generation/utils.py b/src/transformers/generation/utils.py
@@ -2170,7 +2170,7 @@ def _valid_auto_compile_criteria(self, model_kwargs: dict[str, Any], generation_
             return False
 
         # Base logic
-        valid_hardware = self.device.type == "cuda" or bool(
+        valid_hardware = self.device.type in ["cuda", "xpu"] or bool(
             generation_config.compile_config is not None and generation_config.compile_config._compile_all_devices
         )
         using_compilable_cache = (

Original file line number	Diff line number	Diff line change
`@@ -2170,7 +2170,7 @@ def _valid_auto_compile_criteria(self, model_kwargs: dict[str, Any], generation_`
`2170`	`2170`	`return False`
`2171`	`2171`
`2172`	`2172`	`# Base logic`
`2173`		`- valid_hardware = self.device.type == "cuda" or bool(`
	`2173`	`+ valid_hardware = self.device.type in ["cuda", "xpu"] or bool(`
`2174`	`2174`	`generation_config.compile_config is not None and generation_config.compile_config._compile_all_devices`
`2175`	`2175`	`)`
`2176`	`2176`	`using_compilable_cache = (`