[Triton-MLIR][BACKEND] Add elementwise ops and tests (#804)

Co-authored-by: Keren Zhou <kerenzhou@openai.com>
2022-10-28 13:26:29 +08:00
parent 3b80801dff
commit 3685194456
9 changed files with 616 additions and 60 deletions
--- a/python/src/triton.cc
+++ b/python/src/triton.cc
@@ -1335,6 +1335,12 @@ void init_triton_translation(py::module &m) {
          py::bytes bytes(cubin);
          return bytes;
        });
+
+  m.def("add_external_libs",
+        [](mlir::ModuleOp &op, const std::vector<std::string> &names,
+           const std::vector<std::string> &paths) {
+          ::mlir::triton::addExternalLibs(op, names, paths);
+        });
 }

 void init_triton(py::module &m) {