Misc cleanup

triton-lang · Mar 14, 2023 · e01d246 · e01d246
1 parent b0068f0
commit e01d246
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 13 deletions.
diff --git a/python/src/triton.cc b/python/src/triton.cc
@@ -866,18 +866,6 @@ void init_triton_ir(py::module &&m) {
              auto loc = self.getUnknownLoc();
              return self.create<mlir::arith::RemUIOp>(loc, lhs, rhs);
            })
-      .def("create_fmin",
-           [](mlir::OpBuilder &self, mlir::Value &lhs,
-              mlir::Value &rhs) -> mlir::Value {
-             auto loc = self.getUnknownLoc();
-             return self.create<mlir::arith::MinFOp>(loc, lhs, rhs);
-           })
-      .def("create_smin",
-           [](mlir::OpBuilder &self, mlir::Value &lhs,
-              mlir::Value &rhs) -> mlir::Value {
-             auto loc = self.getUnknownLoc();
-             return self.create<mlir::arith::MinSIOp>(loc, lhs, rhs);
-           })
       .def("create_add",
            [](mlir::OpBuilder &self, mlir::Value &lhs,
               mlir::Value &rhs) -> mlir::Value {

diff --git a/python/triton/language/__init__.py b/python/triton/language/__init__.py
@@ -154,7 +154,6 @@
     "philox_impl",
     "pi32_t",
     "pointer_type",
-    "prod",
     "program_id",
     "rand",
     "rand4x",

diff --git a/python/triton/language/core.py b/python/triton/language/core.py
@@ -1121,12 +1121,14 @@ def _decorator(func: T) -> T:
 
     return _decorator
 
+
 @contextmanager
 def _insertion_guard(builder):
     ip = builder.get_insertion_point()
     yield
     builder.restore_insertion_point(ip)
 
+
 @builtin
 def reduction(input, axis, combine_fn, _builder=None, _generator=None):
     """Applies the combine_fn to all elements in :code:`input` tensors along the provided :code:`axis`