Don't set the promotion state explicitly.

The method `_set_promotion_state` was removed in numpy 2.2 and the promotion state is set to weak by default: https://numpy.org/devdocs/release/2.2.0-notes.html#nep-50-promotion-state-option-removed PiperOrigin-RevId: 714013325
openxla · Jan 10, 2025 · 6a357aa · 6a357aa
1 parent 69e993c
commit 6a357aa
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/third_party/triton/temporary/numpy_type_promotion.patch b/third_party/triton/temporary/numpy_type_promotion.patch
@@ -0,0 +1,12 @@
+--- a/python/test/unit/language/test_core.py
++++ b/python/test/unit/language/test_core.py
+@@ -363,8 +363,7 @@ def _test_binary(dtype_x, dtype_y, expr,
+             # We remove any explicit casting
+             pattern = r'\.astype\(np\.\w+\)'
+             scalar_expr = expr if numpy_expr is None else re.sub(pattern, '', numpy_expr)
+-            with promotion_numpy_2_0():
+-                z_ref = eval(scalar_expr)
++            z_ref = eval(scalar_expr)
+         else:
+             z_ref = eval(expr if numpy_expr is None else numpy_expr)
+
diff --git a/third_party/triton/temporary/series.bzl b/third_party/triton/temporary/series.bzl
@@ -14,5 +14,6 @@ those to this list.
 """
 
 temporary_patch_list = [
+    "//third_party/triton:temporary/numpy_type_promotion.patch",
     # Add new patches just above this line
 ]