Update on "[inductor] add lowering for repeat_interleave.Tensor with output size specified (#147160)"

v0i0 · v0i0 · commit 591691b85a74 · 2025-07-25T16:58:14.000-07:00
cc voznesenskym penguinwu EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng wenzhe-nrv jiayisunx ipiszy chenyang78 kadeng muchulee8 amjames chauhang aakhundov coconutruben

[ghstack-poisoned]
diff --git a/torch/_inductor/decomposition.py b/torch/_inductor/decomposition.py
@@ -1163,7 +1163,7 @@ def repeat_interleave_Tensor(
         return NotImplemented
     if output_size is None or type(output_size) is not int:
         return NotImplemented
-    if repeat.device == "mps":
+    if repeat.device.type == "mps":
         return NotImplemented
     assert repeat.dtype in [torch.int32, torch.int64]
     assert repeat.ndim == 1
