change test_fp16 -> _run

Author: DrRyanHuang

paddle/cinn/hlir/dialect/operator/transforms/add_cast_to_elementwise_add_pass.cc

@@ -27,13 +27,6 @@ namespace cinn {
 namespace dialect {
 namespace ir {
 
-auto is_integer_or_bool = [](const pir::Type& x) {
-  return x.isa<pir::IndexType>() || x.isa<pir::Int64Type>() ||
-         x.isa<pir::Int32Type>() || x.isa<pir::Int16Type>() ||
-         x.isa<pir::Int8Type>() || x.isa<pir::UInt8Type>() ||
-         x.isa<pir::BoolType>();
-};
-
 pir::Type GetOutputDtype(const pir::Type& x, const pir::Type& y) {
   pir::IrContext* context = pir::IrContext::Instance();
   // type promotion
@@ -44,6 +37,13 @@ pir::Type GetOutputDtype(const pir::Type& x, const pir::Type& y) {
     return pir::Complex64Type::get(context);
   }
 
+  auto is_integer_or_bool = [](const pir::Type& x) {
+    return x.isa<pir::IndexType>() || x.isa<pir::Int64Type>() ||
+           x.isa<pir::Int32Type>() || x.isa<pir::Int16Type>() ||
+           x.isa<pir::Int8Type>() || x.isa<pir::UInt8Type>() ||
+           x.isa<pir::BoolType>();
+  };
+
   if (is_integer_or_bool(x) || is_integer_or_bool(y)) {
     PADDLE_THROW(::common::errors::InvalidType(
         "Type promotion only support calculations between floating-point "

paddle/cinn/hlir/dialect/operator/transforms/add_cast_to_elementwise_add_pass.h

@@ -1,4 +1,4 @@
-// Copyright (c) 2023 PaddlePaddle Authors. All Rights Reserved.
+// Copyright (c) 2025 PaddlePaddle Authors. All Rights Reserved.
 //
 // Licensed under the Apache License, Version 2.0 (the "License");
 // you may not use this file except in compliance with the License.

test/dygraph_to_static/test_cast_pass.py

@@ -1,4 +1,4 @@
-# Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved.
+# Copyright (c) 2025 PaddlePaddle Authors. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -32,16 +32,19 @@ def func(x, y):
 
 class TestAddCastToElementwiseAddPass(Dy2StTestBase):
     # test AddCastToElementwiseAddPass
-    def test_bf16(self, dtype="float16"):
+    def _run(self, dtype):
         static_fn = paddle.jit.to_static(func)
         x = paddle.randn([200, 200])
         y = paddle.randn([200, 200], dtype=dtype)
         np.testing.assert_allclose(
             static_fn(x, y).numpy(), x.numpy() + y.cast("float32").numpy()
         )
 
+    def test_bf16(self):
+        self._run(dtype="bfloat16")
+
     def test_fp16(self):
-        self.test_bf16("float16")
+        self._run(dtype="float16")
 
 
 if __name__ == '__main__':