From eeedeef3b84236ad8effdbd56f67b220fda4768a Mon Sep 17 00:00:00 2001
From: maochuanjun <10255501521@stu.ecnu.edu.cn>
Date: Wed, 27 May 2026 21:48:19 +0800
Subject: [PATCH] Support acos, atan, and atan2 ops via patches

---
 patches/0003-add-trig-ops-to-frontend.patch | 59 +++++++++++++++++++++
 1 file changed, 59 insertions(+)
 create mode 100644 patches/0003-add-trig-ops-to-frontend.patch

diff --git a/patches/0003-add-trig-ops-to-frontend.patch b/patches/0003-add-trig-ops-to-frontend.patch
new file mode 100644
index 00000000..66d8ef9e
--- /dev/null
+++ b/patches/0003-add-trig-ops-to-frontend.patch
@@ -0,0 +1,59 @@
+diff --git a/python/src/ir.cc b/python/src/ir.cc
+index 4c8a4233b..076e96bd0 100644
+--- a/python/src/ir.cc
++++ b/python/src/ir.cc
+@@ -1614,6 +1614,18 @@ void init_triton_ir(py::module &&m) {
+            [](TritonOpBuilder &self, Value &val) -> Value {
+              return self.create<math::SinOp>(val);
+            })
++      .def("create_acos",
++           [](TritonOpBuilder &self, Value &val) -> Value {
++             return self.create<math::AcosOp>(val);
++           })
++      .def("create_atan",
++           [](TritonOpBuilder &self, Value &val) -> Value {
++             return self.create<math::AtanOp>(val);
++           })
++      .def("create_atan2",
++           [](TritonOpBuilder &self, Value &lhs, Value &rhs) -> Value {
++             return self.create<math::Atan2Op>(lhs, rhs);
++           })
+       .def("create_log",
+            [](TritonOpBuilder &self, Value &val) -> Value {
+              return self.create<math::LogOp>(val);
+diff --git a/python/triton/language/math.py b/python/triton/language/math.py
+index 582cd876c..f74a20ebd 100644
+--- a/python/triton/language/math.py
++++ b/python/triton/language/math.py
+@@ -247,3 +247,31 @@ def fma(x, y, z, _semantic=None):
+     z, x = core.binary_op_type_legalization(z, x, _semantic)
+     z, y = core.binary_op_type_legalization(z, y, _semantic)
+     return core.tensor(_semantic.builder.create_fma(x.handle, y.handle, z.handle), x.type)
++
++
++@core.builtin
++@_check_dtype(dtypes=["fp32", "fp64"])
++@_add_math_1arg_docstr("inverse cosine")
++@core._tensor_member_fn
++def acos(x, _semantic=None):
++    x = _semantic.to_tensor(x)
++    return core.tensor(_semantic.builder.create_acos(x.handle), x.type)
++
++
++@core.builtin
++@_check_dtype(dtypes=["fp32", "fp64"])
++@_add_math_1arg_docstr("inverse tangent")
++@core._tensor_member_fn
++def atan(x, _semantic=None):
++    x = _semantic.to_tensor(x)
++    return core.tensor(_semantic.builder.create_atan(x.handle), x.type)
++
++
++@core.builtin
++@_check_dtype(dtypes=["fp32", "fp64"])
++@_add_math_2arg_docstr("four-quadrant inverse tangent")
++def atan2(x, y, _semantic=None):
++    x = _semantic.to_tensor(x)
++    y = _semantic.to_tensor(y)
++    x, y = core.binary_op_type_legalization(x, y, _semantic)
++    return core.tensor(_semantic.builder.create_atan2(x.handle, y.handle), x.type)