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

v0i0 · v0i0 · commit ef2360707847 · 2025-07-16T18:00:07.000-07:00
[ghstack-poisoned]
diff --git a/test/inductor/test_torchinductor.py b/test/inductor/test_torchinductor.py
@@ -13626,7 +13626,6 @@ def forward(self, x):
                 FileCheck().check("cpp_fused_add_0").run(code)
             self.assertEqual(refe_out, test_out)
 
-
     @parametrize("dtype", [torch.int32, torch.int64])
     def test_repeat_interleave_Tensor_decomp(self, dtype):
         device = "cpu"
diff --git a/torch/_inductor/decomposition.py b/torch/_inductor/decomposition.py
@@ -1163,4 +1163,6 @@ def repeast_interleave_Tensor(
         return NotImplemented
     cumsum = repeat.cumsum(0)
     pos = torch.arange(output_size, device=repeat.device)
-    return torch.searchsorted(cumsum, pos, out_int32=(repeat.dtype == torch.int32), right=True)
+    return torch.searchsorted(
+        cumsum, pos, out_int32=(repeat.dtype == torch.int32), right=True
+    )
