[CINN] Fix int type casting of Scale op (#72811)

lshpku · web-flow · commit fc24bc7a7ad6 · 2025-05-21T16:18:59.000+08:00
diff --git a/paddle/cinn/hlir/op/elementwise.cc b/paddle/cinn/hlir/op/elementwise.cc
@@ -326,10 +326,10 @@ std::shared_ptr<OpStrategy> StrategyForScaleSymbolic(
 
               Expr cast_scale = should_upscale_fp32
                                     ? Expr(scale)
-                                    : ir::Cast::Make(A->type(), Expr(scale));
+                                    : common::cast(Expr(scale), A->type());
               Expr cast_bias = should_upscale_fp32
                                    ? Expr(bias)
-                                   : ir::Cast::Make(A->type(), Expr(bias));
+                                   : common::cast(Expr(bias), A->type());
               Expr add_result;
               if (scale == 1.0f) {
                 if (bias == 0.0f) {
diff --git a/test/ir/pir/cinn/test_cinn_arange.py b/test/ir/pir/cinn/test_cinn_arange.py
@@ -65,7 +65,9 @@ def func():
 
     def test_broadcast_tensor(self):
         def func():
-            return paddle.arange(1, 2) + paddle.arange(end=32769)
+            a = paddle.arange(1, 2, dtype="int32")
+            b = paddle.arange(end=32769)
+            return (a * 3).astype("int64") + b
 
         self.eval(func, [])
 
